enable find_unused_parameters flag for DDP

2023-11-21 19:26:42 +08:00
parent 112208ef74
commit 388974ab2a
3 changed files with 4 additions and 3 deletions
@@ -46,6 +46,7 @@ scheduler_cfg:
  scheduler: MultiStepLR

 trainer_cfg:
+  find_unused_parameters: false
  enable_float16: true
  with_test: false
  fix_BN: false