facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1.402 geöffnet am 30. Okt. 2017

Auf GitHub ansehen
 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Shell (1.954 Forks)batch import
enhancementhelp wanted

Repository-Metriken

Stars
 (8.424 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

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.

Contributor Guide