From e9d34320e06f0b3598d08e03f997c2117a5544f1 Mon Sep 17 00:00:00 2001
From: Mike Walmsley <walmsleymk1@gmail.com>
Date: Wed, 22 Nov 2023 14:02:28 -0500
Subject: [PATCH] v2

---
 only_for_me/narval/train.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
index dda8efba..225191ec 100644
--- a/only_for_me/narval/train.sh
+++ b/only_for_me/narval/train.sh
@@ -24,7 +24,7 @@ srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
     --save-dir $REPO_DIR/only_for_me/narval/desi_f128 \
     --batch-size 256 \
     --num-features 128 \
-    --gpus 2 \
+    --gpus 1 \
     --num-workers 10 \
     --color --wandb --mixed-precision --compile-encoder