未验证 提交 ccd15f51 编写于 作者: G gaotingquan

uncommit: sync bn is too slow to use and convert_sync_batchnorm() is not effective for BatchNorm

上级 5ca50f84
...@@ -243,11 +243,6 @@ class Engine(object): ...@@ -243,11 +243,6 @@ class Engine(object):
level=amp_level, level=amp_level,
save_dtype='float32') save_dtype='float32')
# TODO(gaotingquan): convert_sync_batchnorm is not effective
# eval loss in training is inconsistent with the eval only if bn is used,
# because the running_mean and running_var of bn are not synced in dist.
self.model = nn.SyncBatchNorm.convert_sync_batchnorm(self.model)
# for distributed # for distributed
world_size = dist.get_world_size() world_size = dist.get_world_size()
self.config["Global"]["distributed"] = world_size != 1 self.config["Global"]["distributed"] = world_size != 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册