Add nn.SyncBatchNorm in ImageNet example

Description

Guia do colaborador

Pilha de tecnologia

python

Domain

machine learning

Tipo Issue

feature

DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.

3

Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.

1-3 hours

Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.

stale

ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.

unclear

Prerequisites

Basic knowledge of PyTorchFamiliarity with ImageNet exampleUnderstanding of SyncBatchNorm

Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.

35

Direção de pesquisa

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.