Add nn.SyncBatchNorm in ImageNet example

Description

Guide contributeur

Stack technique

python

Domaine

machine learning

Type d'issue

feature

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.

1-3 hours

Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.

stale

ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.

unclear

Prérequis

Basic knowledge of PyTorchFamiliarity with ImageNet exampleUnderstanding of SyncBatchNorm

Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.

35

Direction de recherche

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.