diff --git a/main.py b/main.py
index b64ed00..f32fce0 100644
--- a/main.py
+++ b/main.py
@@ -655,7 +655,9 @@ if __name__ == "__main__":
             del callbacks_cfg['ignore_keys_callback']
 
         trainer_kwargs["callbacks"] = [instantiate_from_config(callbacks_cfg[k]) for k in callbacks_cfg]
-
+        trainer_kwargs["plugins"] = list()
+        from pytorch_lightning.plugins import DDPPlugin
+        trainer_kwargs["plugins"].append(DDPPlugin(find_unused_parameters=False))
         trainer = Trainer.from_argparse_args(trainer_opt, **trainer_kwargs)
         trainer.logdir = logdir  ###