FluxML/Flux.jl

Allow BatchNorm training on CUDA with `track_stats=false`

Fermée

#1 606 ouverte le 31 mai 2021

Voir sur GitHubGuide contributeur
 (4 commentaires) (0 réaction) (0 personne assignée)Julia (619 forks)batch import
good first issuehelp wanted

Métriques du dépôt

Stars
 (4 725 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

Gathered from https://discourse.julialang.org/t/batchnorm-only-track-stats-true-supported-on-gpu/62091.

This would most likely require changes in NNlibCUDA as well. I'm not sure how interchangeable the various cudnnBatchNormalizationForward* functions are, so putting a pin in this until someone more knowledgeable can comment.

Guide contributeur

Direction de recherche
Examinez l'implémentation de BatchNorm dans Flux.jl et NNlibCUDA.jl. Comprenez les fonctions cudnnBatchNormalizationForward et modifiez les pour prendre en charge track stats=false. Testez les modifications sur un GPU compatible CUDA.
Stack technique
Aucun
Domaine
backendmachine 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.
3
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.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
JuliaCUDAGit
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30