facebookarchive/caffe2

MKL-DNN integration does not support backward pass

Open

#2376 aperta il 22 mar 2018

Vedi su GitHub
 (1 commento) (0 reazioni) (0 assegnatari)Shell (1954 fork)batch import
help wanted

Metriche repository

Star
 (8424 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

MKL-DNN integration (in caffe2/mkl) implements only forward pass operators, although MKL-DNN itself provides primitives for both forward and backward pass. Implementation of ConvGradient operator would let Caffe2 use MKL-DNN for training on CPU, and support some models which need backprop (e.g. deep dream).

Guida contributor

Direzione di ricerca
Esamina l'implementazione esistente del passaggio in avanti nella directory caffe2/mkl per comprenderne la struttura degli operatori. Studia la documentazione di MKL DNN per le primitive del passaggio all'indietro, in particolare per la convoluzione. Esamina i pattern degli operatori di gradiente altrove in Caffe2 (ad esempio, operatori CPU o CUDA) per definire ConvGradient. Proponi un piano con file specifici da modificare e test da aggiungere.
Tech stack
cpppythonshell
Dominio
machine learningai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
4
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
3-5 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Datata
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Knowledge of MKL DNN primitiveC++ development experienceFamiliarity with Caffe2 operator registration
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
15