facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1402 aperta il 30 ott 2017

Vedi su GitHub
 (1 commento) (0 reazioni) (0 assegnatari)Shell (1954 fork)batch import
enhancementhelp wanted

Metriche repository

Star
 (8424 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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.

Guida contributor