Commit cd5a1e13 authored by Marta's avatar Marta
Browse files

dont try to use ddp for single-gpu single-node training

parent 35449bf1
......@@ -159,8 +159,10 @@ def main(args):
if(args.deepspeed_config_path is not None):
strategy = DeepSpeedPlugin(config=args.deepspeed_config_path)
else:
elif args.gpus > 1 or args.num_nodes > 1:
strategy = "ddp"
else:
strategy = None
trainer = pl.Trainer.from_argparse_args(
args,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment