forked from xmartlabs/spoter-embeddings
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.sh
executable file
·24 lines (24 loc) · 682 Bytes
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/bin/sh
python -m train \
--save_checkpoints_every -1 \
--experiment_name "augment_rotate_75_x8" \
--epochs 10 \
--optimizer "SGD" \
--lr 0.001 \
--batch_size 32 \
--dataset_name "wlasl" \
--training_set_path "WLASL100_train.csv" \
--validation_set_path "WLASL100_test.csv" \
--vector_length 32 \
--epoch_iters -1 \
--scheduler_factor 0 \
--hard_triplet_mining "in_batch" \
--filter_easy_triplets \
--triplet_loss_margin 1 \
--dropout 0.2 \
--start_mining_hard=200 \
--hard_mining_pre_batch_multipler=16 \
--hard_mining_pre_batch_mining_count=5 \
--augmentations_prob=0.75 \
--hard_mining_scheduler_triplets_threshold=0 \
# --normalize_embeddings \