repair train bug in multi gpu

Author: liuhongen1234567

configs/rec/rec_latex_ocr.yml

@@ -19,6 +19,7 @@ Global:
   rec_char_dict_path:  ppocr/utils/dict/latex_ocr_tokenizer.json
   save_res_path: ./output/rec/predicts_latexocr.txt
   d2s_train_image_shape: [1,256,256]
+  find_unused_parameters: True
 
 Optimizer:
   name: AdamW

tools/train.py

@@ -217,7 +217,10 @@ def main(config, device, logger, vdl_writer, seed):
     )
 
     if config["Global"]["distributed"]:
-        model = paddle.DataParallel(model)
+        find_unused_parameters = config["Global"].get("find_unused_parameters", False)
+        model = paddle.DataParallel(
+            model, find_unused_parameters=find_unused_parameters
+        )
     # start train
     program.train(
         config,