Matching main for ddp

2022-09-26 12:44:22 -05:00 · 2022-09-26 12:44:22 -05:00 · ee5b27386f
parent 67154ad9b5
commit ee5b27386f
1 changed files with 0 additions and 2 deletions
--- a/main.py
+++ b/main.py
@ -622,8 +622,6 @@ if __name__ == "__main__":
        # Set the steps
        trainer_config.max_steps = opt.batch_size

-        # default to ddp
-        trainer_config["accelerator"] = "ddp"
        for k in nondefault_trainer_args(opt):
            trainer_config[k] = getattr(opt, k)
        if not "gpus" in trainer_config: