typo

mwalmsley · Nov 20, 2023 · dba85dc · dba85dc
1 parent 42e18f2
commit dba85dc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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