Skip to content

GitLab

"tools/utils/array_readwriter/numpy_array.py" did not exist on "60bc0b7692c6733dd930cd4a502f4171720da38d"
Commit eb54efa2 authored by Binh Tang's avatar Binh Tang Committed by Facebook GitHub Bot
Browse files

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 7778f667
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/runner/test_runner_lightning_quantization.py
View file @ eb54efa2
...@@ -189,7 +189,7 @@ class TestQuantizationAwareTraining(unittest.TestCase): ...@@ -189,7 +189,7 @@ class TestQuantizationAwareTraining(unittest.TestCase):
num_epochs = 2 num_epochs = 2
qat = QuantizationAwareTraining() qat = QuantizationAwareTraining()
trainer = Trainer( trainer = Trainer(
accelerator="ddp_cpu", accelerator="cpu",
num_processes=1, num_processes=1,
default_root_dir=os.path.join(root_dir, "quantized"), default_root_dir=os.path.join(root_dir, "quantized"),
checkpoint_callback=False, checkpoint_callback=False,
......
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