facebookarchive/caffe2
Accuracy requires INT32 while OneHot requires INT64
オープン
#1,402 opened on 2017/10/30
enhancementhelp wanted
Repository metrics
- Stars
- (8,424 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
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.