gotta go fast

2022-09-24 01:13:36 -07:00 · 2022-09-24 01:13:36 -07:00 · aa91eaf228
parent ae05402bc6
commit aa91eaf228
1 changed files with 3 additions and 1 deletions
--- 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  ###