update train.py

2022-08-10 11:04:06 +00:00 · 2022-08-10 11:04:06 +00:00 · 4ac17fca55
parent 70145c3cdb
commit 4ac17fca55
1 changed files with 0 additions and 2 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -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)