Use inferred compute config in Clifford

hlinander · May 26, 2024 · a3ec15b · a3ec15b
1 parent aa6c785
commit a3ec15b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/clifford/clifford.py b/experiments/clifford/clifford.py
@@ -258,7 +258,7 @@ def reg_loss(outputs, batch):
     )
     train_eval = create_regression_metrics(torch.nn.functional.l1_loss, None)
     train_run = TrainRun(
-        compute_config=ComputeConfig(distributed=False, num_workers=10),
+        compute_config=ComputeConfig(),
         train_config=train_config,
         train_eval=train_eval,
         epochs=200,