facebookarchive/caffe2
Voir sur GitHubAccuracy requires INT32 while OneHot requires INT64
Open
#1 402 ouverte le 30 oct. 2017
enhancementhelp wanted
Métriques du dépôt
- Stars
- (8 424 stars)
- Métriques de merge PR
- (Aucune PR mergée en 30 j)
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.