facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1 402 ouverte le 30 oct. 2017

Voir sur GitHub
 (1 commentaire) (0 réactions) (0 assignés)Shell (1 954 forks)batch import
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.

Guide contributeur