4 gpu

mwalmsley · Nov 4, 2023 · 38ad666 · 38ad666
1 parent fa5ec25
commit 38ad666
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/only_for_me/narval/train.py b/only_for_me/narval/train.py
@@ -118,7 +118,7 @@
         resize_after_crop=args.resize_after_crop,
         # hardware parameters
         # gpus=args.gpus,
-        gpus=2,
+        gpus=4,
         nodes=args.nodes,
         mixed_precision=args.mixed_precision,
         wandb_logger=wandb_logger,