facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Aberta

#1.402 aberto em 30 de out. de 2017

 (1 comentário) (0 reação) (0 responsável)Shell (1.954 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (8.424 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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.

Guia do colaborador