Unverified 提交 951922c7 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Add `--sync-bn` known issue (#4032)

* Add `--sync-bn` known issue * Update train.py
上级 b7e985e3
...@@ -217,6 +217,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary ...@@ -217,6 +217,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
# SyncBatchNorm # SyncBatchNorm
if opt.sync_bn and cuda and RANK != -1: if opt.sync_bn and cuda and RANK != -1:
raise Exception('can not train with --sync-bn, known issue https://github.com/ultralytics/yolov5/issues/3998')
model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model).to(device) model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model).to(device)
logger.info('Using SyncBatchNorm()') logger.info('Using SyncBatchNorm()')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论