Some fixes

2023-04-17 15:52:19 +00:00
parent 2e66cccf50
commit 2f7063b70d
6 changed files with 158 additions and 61 deletions
--- a/train.sh
+++ b/train.sh
@@ -1,10 +1,10 @@
 #!/bin/sh
-python -m train \
+python3 -m train \
 	--save_checkpoints_every 10 \
    --experiment_name "wlasl" \
-    --epochs 300 \
-	--optimizer "ADAM" \
-	--lr 0.0001 \
+    --epochs 600 \
+	--optimizer "SGD" \
+	--lr 0.001 \
    --batch_size 16 \
 	--dataset_name "WLASL" \
 	--training_set_path "WLASL100_train.csv" \
@@ -12,7 +12,7 @@ python -m train \
 	--vector_length 32 \
 	--epoch_iters -1 \
 	--scheduler_factor 0.2 \
-	--hard_triplet_mining "None" \
+	--hard_triplet_mining "in_batch" \
 	--filter_easy_triplets \
 	--triplet_loss_margin 2 \
 	--dropout 0.2 \