From 4ac17fca551cc3b2085351842f15c88405cc8d57 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Wed, 10 Aug 2022 11:04:06 +0000
Subject: [PATCH] update train.py

---
 tools/train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index ddd7c312..0c881eca 100755
--- 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)
 
-- 
GitLab