undo device change

openpipelines-bio · Dec 18, 2024 · 856187d · 856187d
1 parent ef1f356
commit 856187d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/integrate/scvi/script.py b/src/integrate/scvi/script.py
@@ -214,8 +214,7 @@ def main():
         early_stopping_min_delta=par["early_stopping_min_delta"],
         plan_kwargs=plan_kwargs,
         check_val_every_n_epoch=1,
-        accelerator="cpu",
-        devices="auto",
+        accelerator="auto",
     )
     # Note: train_size=1.0 should give better results, but then can't do early_stopping on validation set