Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "88fccd145eda3938a83945b64ab6f811afbdd3c6"
Commit 86e2dc7f authored by Ananth Subramaniam's avatar Ananth Subramaniam Committed by Facebook GitHub Bot
Browse files

Update test to not specify Trainer(num_processes) 

Reviewed By: tangbinh

Differential Revision: D34669294

fbshipit-source-id: c87bc1d4c589518f7c9fc21e6dfe27b03e700b6d
parent b2e94a40
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
tests/runner/test_runner_lightning_quantization.py
View file @ 86e2dc7f
...@@ -190,14 +190,12 @@ class TestQuantizationAwareTraining(unittest.TestCase): ...@@ -190,14 +190,12 @@ class TestQuantizationAwareTraining(unittest.TestCase):
qat = QuantizationAwareTraining() qat = QuantizationAwareTraining()
trainer = Trainer( trainer = Trainer(
accelerator="cpu", accelerator="cpu",
num_processes=1, devices=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,
callbacks=[qat], callbacks=[qat],
max_epochs=num_epochs, max_epochs=num_epochs,
logger=False, logger=False,
strategy="ddp_find_unused_parameters_false",
plugins=[],
) )
trainer.fit(model) trainer.fit(model)
......
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