提交 4ac17fca 编写于 作者: 文幕地方's avatar 文幕地方

update train.py

上级 70145c3c
......@@ -123,8 +123,6 @@ def main(config, device, logger, vdl_writer):
if use_sync_bn:
model = paddle.nn.SyncBatchNorm.convert_sync_batchnorm(model)
logger.info('convert_sync_batchnorm')
if config['Global']['distributed']:
model = paddle.DataParallel(model)
model = apply_to_static(model, config, logger)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册