pytorch/vision

Classification references does not work without distributed setup

Open

Aperta il 1 set 2022

Vedi su GitHub
 (2 commenti) (0 reazioni) (0 assegnatari)Python (15.050 star) (6858 fork)batch import
bughelp wantedmodule: reference scripts

Descrizione

If you don't set the respective env vars

https://github.com/pytorch/vision/blob/d5bd8b728f14c33b339fc45c90ca39be339bce3f/references/classification/utils.py#L255-L258

training will not be distributed and in turn the backend will not be initialized. However, during evaluation we check

https://github.com/pytorch/vision/blob/d5bd8b728f14c33b339fc45c90ca39be339bce3f/references/classification/train.py#L88

unguarded, which then fails with

RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.

cc @datumbox

Guida contributor

Tech stack
pythonpytorch
Dominio
machine learningbackend
Tipo issue
bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 hours
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
stale
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
basic Pythonbasic PyTorchunderstanding of torch.distributed
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
70
Direzione di ricerca
The issue is in PyTorch Vision's classification reference scripts. The bug occurs when distributed training env vars are not set, causing an unguarded `dist.*` call in `references/classification/train.py` line 88. The fix should add a check like `if dist.is available() and dist.is initialized()` before using distributed utilities. Also ensure that `utils.py` lines 255-258 properly guard the distributed initialization. See the linked lines in the issue for exact locations.