kaldi-asr/kaldi

Pytorchnn support multi-gpus

Open

#4 468 ouverte le 4 mars 2021

Voir sur GitHub
 (6 commentaires) (0 réactions) (0 assignés)Shell (5 359 forks)batch import
enhancementhelp wanted

Métriques du dépôt

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

Description

Pytorchnn only uses one gpu. It's very inefficient when training a large amount of text.

Guide contributeur

Direction de recherche
La demande consiste à étendre le module Pytorchnn de Kaldi pour utiliser plusieurs GPU. Commencez par examiner l'implémentation actuelle à GPU unique dans le code source (probablement sous src/pytorchnn/). Étudiez les modules DataParallel et DistributedDataParallel de PyTorch. Vérifiez les commentaires de l'issue pour toute approche suggérée ou indication des mainteneurs. Proposez ensuite une conception qui intègre la prise en charge multi GPU sans compromettre les fonctionnalités existantes.
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.
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.
Plutôt claire
Prérequis
CUDAGPU programming
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30