facebookarchive/caffe2
GitHub で見るAccuracy requires INT32 while OneHot requires INT64
Open
#1,402 opened on 2017年10月30日
enhancementhelp wanted
Repository metrics
- Stars
- (8,424 stars)
- PR merge metrics
- (30d に merged PR はありません)
説明
The operator Accuracy() takes parameter labels as type core.DataType.INT32.
OneHot() takes parameter indices as type core.DataType.INT64.
LabelCrossEntropy() takes label as type core.DataType.INT32.
It's a bit waste of time to cast the label back and forth, hope there is alignment. Thx.