facebookarchive/caffe2
Ver no GitHubAccuracy requires INT32 while OneHot requires INT64
Open
#1.402 aberto em 30 de out. de 2017
enhancementhelp wanted
Description
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.