facebookarchive/caffe2

MKL-DNN integration does not support backward pass

Open

#2 376 ouverte le 22 mars 2018

Voir sur GitHub
 (1 commentaire) (0 réactions) (0 assignés)Shell (1 954 forks)batch import
help wanted

Métriques du dépôt

Stars
 (8 424 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

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).

Guide contributeur

Direction de recherche
Étudiez l'implémentation existante du forward pass de l'opérateur Conv dans caffe2/mkl et utilisez les primitives de backward pass de MKL DNN pour implémenter l'opérateur ConvGradient.
Stack technique
cpp
Domaine
machine learning
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
4
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
3-5 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Ancienne
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
C++MKL DNNDeep Learning
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
20