Add nn.SyncBatchNorm in ImageNet example

Repository metrics

Stars

21,634 stars (21,634 stars)

PR merge metrics

No merged PRs in 30d (No merged PRs in 30d)

Description

Contributor guide

Research direction

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.

Tech stack

python

Domain

machine learning

Issue type

Feature

DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.

3

Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.

1-3 hours

Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.

Stale

ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.

Unclear

Prerequisites

Basic knowledge of PyTorchFamiliarity with ImageNet exampleUnderstanding of SyncBatchNorm

Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.

35