Set gpus=0 in examples

2021-05-15 12:43:00 +02:00
parent c639836537
commit ebc42a4aa8
9 changed files with 9 additions and 9 deletions
--- a/examples/glvq_spiral.py
+++ b/examples/glvq_spiral.py
@@ -35,7 +35,7 @@ if __name__ == "__main__":

    # Setup trainer
    trainer = pl.Trainer(
-        gpus=-1,
+        gpus=0,
        max_epochs=200,
        callbacks=[vis, snan],
    )