From aa91eaf228f9fbe3751055b13206258dc9ff4912 Mon Sep 17 00:00:00 2001
From: harubaru <seizou.genkei@protonmail.com>
Date: Sat, 24 Sep 2022 01:13:36 -0700
Subject: [PATCH] gotta go fast

---
 main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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  ###