Script to train base model on full dataset

manolotis · Nov 18, 2022 · ba32980 · ba32980
1 parent 26a84d3
commit ba32980
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/code/bash/rdlm_train_full.sh b/code/bash/rdlm_train_full.sh
@@ -0,0 +1,15 @@
+MODEL_NAME=xception71
+python /home/manolotis/sandbox/robustness_benchmark/motionCNN/train.py \
+    --train-data "/home/manolotis/datasets/waymo/motion v1.0/prerender/motionCNN/training/" \
+    --dev-data "/home/manolotis/datasets/waymo/motion v1.0/prerender/motionCNN/validation/" \
+    --save /home/manolotis/sandbox/robustness_benchmark/motionCNN/trained_models/${MODEL_NAME} \
+    --model ${MODEL_NAME} \
+    --img-res 224 \
+    --in-channels 25 \
+    --time-limit 80 \
+    --n-traj 6 \
+    --lr 0.001 \
+    --batch-size 128 \
+    --n-epochs 1000 \
+    --n-jobs 48 \
+    --n-shards 16