spoterembedding/train.sh
2023-05-21 20:30:12 +00:00

24 lines
664 B
Bash
Executable File

#!/bin/sh
python3 -m train \
--save_checkpoints_every 10 \
--experiment_name "Finetune Fingerspelling Signs" \
--epochs 1000 \
--optimizer "ADAM" \
--lr 0.00001 \
--batch_size 8 \
--dataset_name "FingerSpelling" \
--training_set_path "train.csv" \
--validation_set_path "val.csv" \
--vector_length 32 \
--epoch_iters -1 \
--scheduler_factor 0 \
--hard_triplet_mining "in_batch" \
--filter_easy_triplets \
--start_mining_hard 50 \
--triplet_loss_margin 4 \
--dropout 0.2 \
--tracker=clearml \
--dataset_loader=clearml \
--dataset_project="SpoterEmbedding" \
--finetune \
--checkpoint_path "checkpoints/checkpoint_embed_3835.pth"