facebookarchive/caffe2
Accuracy requires INT32 while OneHot requires INT64
Aberta
#1.402 aberto em 30 de out. de 2017
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.