Commit 1fb10077 authored by Binh Tang's avatar Binh Tang Committed by Facebook GitHub Bot
Browse files

[lightning] Replace deprecated DDP accelerator with ddp_find_unused_parameters_false

Summary: We proactively remove references to the deprecated DDP accelerator to prepare for the breaking changes following the release of PyTorch Lighting 1.6 (see T112240890).

Differential Revision: D34295318

fbshipit-source-id: 7b2245ca9c7c2900f510722b33af8d8eeda49919
parent cbf1b839
...@@ -425,8 +425,7 @@ def cli_main(): ...@@ -425,8 +425,7 @@ def cli_main():
max_epochs=args.epochs, max_epochs=args.epochs,
gpus=args.num_gpu, gpus=args.num_gpu,
num_nodes=args.num_node, num_nodes=args.num_node,
accelerator="ddp", strategy="ddp_find_unused_parameters_false",
plugins=DDPStrategy(find_unused_parameters=False), # make sure there is no unused params
limit_train_batches=1.0, # Useful for fast experiment limit_train_batches=1.0, # Useful for fast experiment
gradient_clip_val=5.0, gradient_clip_val=5.0,
callbacks=callbacks, callbacks=callbacks,
......
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