Added downstream training bash file

epidemicsound · Sep 1, 2023 · bb66956 · bb66956
1 parent 0f9e3c8
commit bb66956
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/downstream_training.sh b/downstream_training.sh
@@ -0,0 +1,10 @@
+set -xe
+python -m src.train_head --name $1 --epochs 50 \
+--dataset nsynth_instrument --accelerator gpu --devices 1 --num_workers 32 --strategy auto \
+--precision 32 --lr 0.3 --weight_decay 0.0001
+python -m src.train_head --name $1 --epochs 50 \
+--dataset nsynth_pitch --accelerator gpu --devices 1 --num_workers 32 --strategy auto \
+--precision 32 --lr 0.03 --weight_decay 0.0001
+python -m src.train_head --name $1 --epochs 100 \
+--dataset mtat --accelerator gpu --devices 1 --num_workers 32 --strategy auto \
+--precision 32 --lr 0.8 --weight_decay 0.0001