-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain_SCST_coco-Med.sh
85 lines (84 loc) · 2.89 KB
/
train_SCST_coco-Med.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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
id="transformer_bushiyan_SRFNet_wanzhengban_Med"
if [ ! -f log/log_$id/infos_$id.pkl ]; then
start_from=""
else
start_from="--start_from log/log_$id"
fi
#python train_1.py --id transformer_bu_k15 \
# --caption_model transformer_bu \
# --refine 1 \
# --refine_aoa 1 \
# --use_ff 0 \
# --decoder_type AoA \
#--use_multi_head 2 \
#--num_heads 8 \
#--multi_head_scale 1 \
#--mean_feats 1 \
#--ctx_drop 1 \
#--dropout_aoa 0.3 \
#--label_smoothing 0.2 \
#--input_json data/coco/cocotalk.json \
#--input_label_h5 data/coco/cocotalk_label.h5 \
#--input_fc_dir data/coco/cocobu_fc \
#--input_att_dir data/coco/cocobu_att \
#--input_word_dir data/coco/cocobu_object_name/num-unk \
#--input_attr_dir data/coco/cocobu_attr_name/num-unk \
#--input_seg_dir /data2/fjq/1.next_paper/coco_seg_results/coco_seg_npy_0.7/num \
#--input_box_dir data/coco/cocobu_box \
#--seq_per_img 5 \
#--batch_size 64 \
#--beam_size 2 \
#--use_box 1 \
#--learning_rate 5e-4 \
#--num_layers 4 \
#--input_encoding_size 512 \
#--rnn_size 2048 \
#--learning_rate_decay_start 0 \
#--scheduled_sampling_start 0 \
#--checkpoint_path log/log_$id \
#$start_from \
#--save_checkpoint_every 2000 \
#--language_eval 1 \
#--val_images_use -1 \
#--max_epochs 15 \
#--scheduled_sampling_increase_every 5 \
#--scheduled_sampling_max_prob 0.5 \
#--learning_rate_decay_every 3
python train_2.py --id $id \
--caption_model transformer_1 \
--refine 1 \
--refine_aoa 1 \
--use_ff 0 \
--decoder_type AoA \
--use_multi_head 2 \
--num_heads 8 \
--multi_head_scale 1 \
--mean_feats 1 \
--ctx_drop 1 \
--dropout_aoa 0.3 \
--input_json data/coco/cocotalk.json \
--input_label_h5 data/coco/cocotalk_label.h5 \
--input_fc_dir /data/fjq/1.next_paper/cocobu_fc \
--input_att_dir /data/fjq/1.next_paper/cocobu_att \
--input_word_dir /data/fjq/1.next_paper/clip_retrieval_filter_new/coco_label_filter_results/cocobu_label_filter_1.0_res101/num \
--input_attr_dir /data/fjq/1.next_paper/clip_retrieval_filter_new/coco_attr_filter_results/cocobu_attr_filter_1.0_res101/num \
--input_seg_dir /data/fjq/1.next_paper/clip_retrieval_filter_new/coco_seg_filter_results/cocobu_seg_filter_0.8_res101/num \
--input_box_dir /data/fjq/1.next_paper/cocobu_box \
--seq_per_img 5 \
--batch_size 32 \
--beam_size 2 \
--num_layers 4 \
--input_encoding_size 256 \
--rnn_size 1024 \
--language_eval 1 \
--val_images_use -1 \
--save_checkpoint_every 2000 \
--start_from log/log_$id \
--checkpoint_path log/log_$id"_rl" \
--learning_rate 5e-6 \
--max_epochs 40 \
--self_critical_after 0 \
--cached_tokens coco-train-idxs \
--learning_rate_decay_start -1 \
--scheduled_sampling_start -1 \
--reduce_on_plateau