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
...@@ -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