-
Notifications
You must be signed in to change notification settings - Fork 0
/
do_predition_iacc.3_ITV.sh
18 lines (12 loc) · 1.42 KB
/
do_predition_iacc.3_ITV.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
rootpath=/vireo00/nikki/AVS_data
testCollection=iacc.3
query_sigmoid_threshold=0.99
logger_name=/vireo00/nikki/AVS_data/tgif-msrvtt10k-VATEX/ACMTOIS/tv2016train/ITV_word_only_dp_0.2_measure_cosine_lambda_0.2/vocab_word_vocab_5_word_dim_500_text_rnn_size_512_text_norm_True_kernel_sizes_2-3-4_num_512/visual_feature_pyresnext-101_rbps13k,flatten0_output,os+pyresnet-152_imagenet11k,flatten0_output,os_visual_rnn_size_1024_visual_norm_True_kernel_sizes_2-3-4-5_num_512/mapping_text_0-2048_img_0-2048_decoder_0-2048/loss_func_mrl_margin_0.2_direction_all_max_violation_True_cost_style_sum/optimizer_adam_lr_0.0002_decay_0.99_grad_clip_2.0_val_metric_recall/run_ITV_on_tgif-msrvtt10k-VATEX
checkpoint_name=model_best.pth.match.tar
#checkpoint_name=model_best.pth.class.tar
overwrite=0
query_sets=tv16.avs.txt,tv17.avs.txt,tv18.avs.txt
query_num_all=90
gpu=0
echo "CUDA_VISIBLE_DEVICES=$gpu python predictor_ITV_batch_querysets.py $testCollection --checkpoint_name $checkpoint_name --query_num_all $query_num_all --query_sigmoid_threshold $query_sigmoid_threshold --rootpath $rootpath --overwrite $overwrite --logger_name $logger_name --query_sets $query_sets"
CUDA_VISIBLE_DEVICES=$gpu python predictor_ITV_batch_querysets.py $testCollection --checkpoint_name $checkpoint_name --query_num_all $query_num_all --query_sigmoid_threshold $query_sigmoid_threshold --rootpath $rootpath --overwrite $overwrite --logger_name $logger_name --query_sets $query_sets