-
Notifications
You must be signed in to change notification settings - Fork 9
/
train_sim10k_final.sh
executable file
·59 lines (59 loc) · 1.49 KB
/
train_sim10k_final.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
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
python scripts/detection/faster_rcnn/train_faster_rcnn_se.py \
--network resnet50_v1b \
--dataset sim10k \
--with_teacher 1 \
--se_alpha 0.999 \
--teacher_aug 1 \
--num_teacher 1 \
--validate_resume 0 \
--use_se 1 \
--augmentation 0 \
--se_rpn_loss 0.0 \
--se_rcnn_loss 1.0 \
--fixed_conf_thres 0 \
--pretrained_base 1 \
--allow_missing 0 \
--ignore_extra 0 \
--pretrained_base_path "../models/" \
--train_patterns '.*dense|.*rpn|.*down(2|3|4)_conv|.*layers(2|3|4)_conv' \
--classes "car" \
--train_root /export/home/v-qcaii/cv19/data \
--val_root /export/home/v-qcaii/cv19/data \
--target_root /export/home/v-qcaii/cv19/data \
--train_split sim10k_trainval.txt \
--val_split cityscape_val.txt \
--target_split cityscape_train.txt \
--min_dataset_size 3000 \
--num-workers 8 \
--gpus 0,1,2,3 \
--epochs 8 \
--start-epoch 0 \
--lr-decay 0.1 \
--lr-decay-epoch 4,6 \
--lr-warmup 1000 \
--momentum 0.9 \
--log-interval 10 \
--save-interval 1 \
--val-interval 1 \
--seed 233 \
--student_se_loss 0 \
--teacher_agree 0 \
--similarity_weight 1.0 \
--similarity_distance_metric "l2" \
--similarity_metric "cosine" \
--similarity_feature "visual" \
--post_softmax 0 \
--inside_graph_loss_weight 1.0 \
--early_stop 0.37 \
--hybrid 0 \
--linear_conf 0 \
--base_conf_thres 0.97 \
--conf_decay_epoch 9 \
--wd 0.0005 \
--lr 0.0001 \
--resume "models-sim10k/sim10k_pretrain.params" \
--save-prefix models-sim10k/ssd_2 \
--similarity_mask_with_equal_label 1 \
--distance_based_inside_graph 1 \
--continue_conf_thres 1 \
--max_conf_thres 1.0