diff --git a/only_for_me/narval/train.py b/only_for_me/narval/train.py
index 105b6155..5993af4b 100644
--- a/only_for_me/narval/train.py
+++ b/only_for_me/narval/train.py
@@ -66,7 +66,7 @@
 
     all_urls = glob.glob(search_str)
     assert len(all_urls) > 0, search_str
-    train_urls, val_urls = all_urls[:8], all_urls[8:]
+    train_urls, val_urls = all_urls[:25], all_urls[25:]
     schema = schemas.decals_all_campaigns_ortho_schema
 
     # debug mode
@@ -99,6 +99,7 @@
         epochs=epochs,  # rely on early stopping
         patience=10,
         # augmentation parameters
+        # color=args.color,
         color=args.color,
         resize_after_crop=args.resize_after_crop,
         # hardware parameters
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
index 9cd30663..3a52382f 100644
--- a/only_for_me/narval/train.sh
+++ b/only_for_me/narval/train.sh
@@ -23,7 +23,7 @@ REPO_DIR=/project/def-bovy/walml/zoobot
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
     --save-dir $REPO_DIR/only_for_me/narval/debug_models \
     --batch-size 128 \
-    --gpus 1
+    --gpus 1 \
     --color --wandb --mixed-precision
 # srun python $SLURM_TMPDIR/zoobot/only_for_me/narval/finetune.py