Since pytorch has supported this feature, why not add it?
Guida contributor
Tech stack
python
Dominio
machine learning
Tipo issue
feature
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.
unclear
Prerequisiti
Basic knowledge of PyTorchFamiliarity with ImageNet exampleUnderstanding of SyncBatchNorm
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
35
Direzione di ricerca
Review the main training script in the ImageNet example (likely train.py or main.py) to identify where nn.BatchNorm is currently used. Then investigate how to integrate nn.SyncBatchNorm from PyTorch's documentation. Note that the issue lacks details on which specific layer to replace or any configuration needed, so further clarification from maintainers may be required.