Files
spoterembedding/train.sh

23 lines
623 B
Bash
Executable File

#!/bin/sh
python3 -m train \
--save_checkpoints_every 1 \
--experiment_name "Finetune Basic Signs" \
--epochs 100 \
--optimizer "ADAM" \
--lr 0.00001 \
--batch_size 16 \
--dataset_name "BasicSigns" \
--training_set_path "train.csv" \
--validation_set_path "val.csv" \
--vector_length 32 \
--epoch_iters -1 \
--scheduler_factor 0.05 \
--hard_triplet_mining "None" \
--filter_easy_triplets \
--triplet_loss_margin 2 \
--dropout 0.2 \
--tracker=clearml \
--dataset_loader=clearml \
--dataset_project="SpoterEmbedding" \
--finetune \
--checkpoint_path "checkpoints/checkpoint_embed_3006.pth"