facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1,402 opened on 2017年10月30日

GitHub で見る
 (1 comment) (0 reactions) (0 assignees)Shell (1,954 forks)batch import
enhancementhelp wanted

Repository metrics

Stars
 (8,424 stars)
PR merge metrics
 (30d に merged PR はありません)

説明

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.

コントリビューターガイド