facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1.402 aberto em 30 de out. de 2017

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)Shell (1.954 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (8.424 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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