facebookarchive/caffe2

Accuracy requires INT32 while OneHot requires INT64

Open

#1.402 geöffnet am 30. Okt. 2017

Auf GitHub ansehen
 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Shell (8.424 Stars) (1.954 Forks)batch import
enhancementhelp wanted

Beschreibung

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.

Contributor Guide

Tech Stack
pythoncpp
Domain
machine learning
Issue Type
feature
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-3 hours
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
stale
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
clear
Voraussetzungen
understanding of Caffe2 operator API
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
50
Research-Richtung
The Accuracy operator expects INT32 labels while OneHot expects INT64 indices, causing a type mismatch. To resolve, examine the source files for AccuracyOp, OneHotOp, and LabelCrossEntropyOp in the caffe2/operators directory to understand their label parameter types. Then propose a consistent type (e.g., change OneHot to use INT32) and update the corresponding tests and documentation. The single comment on the issue may provide additional context.