kaldi-asr/kaldi

Pytorchnn support multi-gpus

Open

#4468 aperta il 4 mar 2021

Vedi su GitHub
 (6 commenti) (0 reazioni) (0 assegnatari)Shell (5359 fork)batch import
enhancementhelp wanted

Metriche repository

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

Descrizione

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

Guida contributor

Direzione di ricerca
La richiesta è di estendere il modulo Pytorchnn di Kaldi per utilizzare più GPU. Inizia esaminando l'implementazione attuale a singola GPU nel codice sorgente (probabilmente in src/pytorchnn/). Studia i moduli DataParallel e DistributedDataParallel di PyTorch. Controlla i commenti della issue per eventuali approcci suggeriti o indicazioni dai manutentori. Quindi propone un progetto che integri il supporto multi GPU senza interrompere le funzionalità esistenti.
Tech stack
Nessuno
Dominio
backendmachine learning
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.
Abbastanza chiara
Prerequisiti
CUDAGPU programming
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
30