Since pytorch has supported this feature, why not add it?
貢獻者指南
技術棧
python
領域
machine learning
議題類型
feature
難度面向新貢獻者的預計實作難度,1 表示很小改動,5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 hours
活動狀態議題目前的可參與程度:新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
unclear
前置要求
Basic knowledge of PyTorchFamiliarity with ImageNet exampleUnderstanding of SyncBatchNorm
新手友善度1-100 的估計分數,表示該議題對首次貢獻者的友善程度。
35
研究方向
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.