-
Notifications
You must be signed in to change notification settings - Fork 8
/
opt_ours_finetuned.txt
28 lines (28 loc) · 1.56 KB
/
opt_ours_finetuned.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
--batch_size 4
--checkpoints_dir checkpoint
--ckpt_path checkpoint/kpsfr.pth
--custom_worldcup_root dataset/WorldCup_2014_2018
--sfp_finetuned False
--gpu_ids 0
--isTrain True
--loss_mode all
--model_archi KC
--name example/opt_ours_finetuned
--nms_thres 0.995
--noise_rotate 0.0084
--noise_trans 5.0
--num_objects 4
--pr_thres 5.0
--public_worldcup_root dataset/soccer_worldcup_2014/soccer_data
--resume False
--step_size 30
--template_path assets
--testset test
--train_epochs 100
--train_lr 0.0001
--train_stage 1
--trainset train
--use_non_local 1
--weight_decay 0.0
--target_video
--target_image