Berkeley-Data / hpt

MIT License
2 stars 3 forks source link

hyperparameter tuning for finetune #63

Closed taeil closed 3 years ago

taeil commented 3 years ago

the best model so far by @suryatechie

great job!

classification/main_train.py --exp_name finetune --data_dir data/sen12ms --label_split_dir splits --IGBP_simple --label_type multi_label --threshold 0.1 --model Moco --lr 0.00001 --decay 1e-5 --batch_size 64 --num_workers 4 --data_size 4096 --epochs 100 --pt_name vivid-resonance-73 --pt_type qe --pt_dir data/pretrained/moco --eval --use_s2 --use_s1