(ms_sl) root@acd91bbbb0c2:/home/ms-sl-main# ./do_tvr.sh $RUN_ID $ROOTPATH $GPU_DEVICE_ID
2023-06-22 07:59:10.549:INFO:main - Setup config, data and model...
------------ Options -------------
{'bsz': '128', 'ckpt_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/model.ckpt', 'clip_scale_w': '0.7', 'collection': 'tvr', 'debug': 'False', 'device': '0', 'device_ids': '[0]', 'drop': '0.1', 'dset_name': 'tvr', 'eval_context_bsz': '200', 'eval_log_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/eval.log.txt', 'eval_path': 'None', 'eval_query_bsz': '50', 'eval_split_name': 'val', 'eval_untrained': 'False', 'exp_id': 'runs_0', 'frame_scale_w': '0.3', 'grad_clip': '-1', 'hard_negative_start_epoch': '20', 'hard_pool_size': '20', 'hidden_size': '384', 'initializer_range': '0.02', 'input_drop': '0.1', 'lr': '0.00025', 'lr_warmup_proportion': '0.01', 'map_size': '32', 'margin': '0.1', 'max_ctx_l': '128', 'max_desc_l': '30', 'max_es_cnt': '10', 'max_position_embeddings': '300', 'max_sub_l': '50', 'model_name': 'MS-SL_Net', 'n_epoch': '100', 'n_heads': '4', 'no_core_driver': 'False', 'no_norm_tfeat': 'False', 'no_norm_vfeat': 'False', 'no_pin_memory': 'False', 'num_workers': '8', 'q_feat_size': '768', 'results_dir': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10', 'results_root': 'results', 'root_path': '/root/VisualSearch', 'seed': '2018', 'sub_feat_size': '768', 'tensorboard_log_dir': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/tensorboard_log', 'train_log_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/train.log.txt', 'train_path': 'None', 'train_span_start_epoch': '0', 'use_sub': 'False', 'vid_feat_size': 'None', 'visual_feature': 'i3d_resnet', 'wd': '0.01'}
[BigFile] 1110332x3072 instances loaded from /root/VisualSearch/tvr/FeatureData/i3d_resnet
2023-06-22 07:59:31.275:INFO:main - model_config {'visual_input_size': 3072, 'query_input_size': 768, 'hidden_size': 384, 'max_ctx_l': 128, 'max_desc_l': 30, 'map_size': 32, 'input_drop': 0.1, 'device': [0], 'drop': 0.1, 'n_heads': 4, 'initializer_range': 0.02, 'margin': 0.1, 'use_hard_negative': False, 'hard_pool_size': 20}
Traceback (most recent call last):
File "method/train.py", line 264, in
model_dir, eval_split_name, eval_path, debug, model_name = start_training(opt)
File "method/train.py", line 251, in start_training
model = NAME_TO_MODELSopt.model_name
KeyError: 'MS-SL_Net'
(ms_sl) root@acd91bbbb0c2:/home/ms-sl-main# ./do_tvr.sh $RUN_ID $ROOTPATH $GPU_DEVICE_ID
2023-06-22 07:59:10.549:INFO:main - Setup config, data and model...
------------ Options -------------
{'bsz': '128', 'ckpt_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/model.ckpt', 'clip_scale_w': '0.7', 'collection': 'tvr', 'debug': 'False', 'device': '0', 'device_ids': '[0]', 'drop': '0.1', 'dset_name': 'tvr', 'eval_context_bsz': '200', 'eval_log_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/eval.log.txt', 'eval_path': 'None', 'eval_query_bsz': '50', 'eval_split_name': 'val', 'eval_untrained': 'False', 'exp_id': 'runs_0', 'frame_scale_w': '0.3', 'grad_clip': '-1', 'hard_negative_start_epoch': '20', 'hard_pool_size': '20', 'hidden_size': '384', 'initializer_range': '0.02', 'input_drop': '0.1', 'lr': '0.00025', 'lr_warmup_proportion': '0.01', 'map_size': '32', 'margin': '0.1', 'max_ctx_l': '128', 'max_desc_l': '30', 'max_es_cnt': '10', 'max_position_embeddings': '300', 'max_sub_l': '50', 'model_name': 'MS-SL_Net', 'n_epoch': '100', 'n_heads': '4', 'no_core_driver': 'False', 'no_norm_tfeat': 'False', 'no_norm_vfeat': 'False', 'no_pin_memory': 'False', 'num_workers': '8', 'q_feat_size': '768', 'results_dir': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10', 'results_root': 'results', 'root_path': '/root/VisualSearch', 'seed': '2018', 'sub_feat_size': '768', 'tensorboard_log_dir': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/tensorboard_log', 'train_log_filepath': '/root/VisualSearch/tvr/results/tvr-runs_0-2023_06_22_07_59_10/train.log.txt', 'train_path': 'None', 'train_span_start_epoch': '0', 'use_sub': 'False', 'vid_feat_size': 'None', 'visual_feature': 'i3d_resnet', 'wd': '0.01'}
[BigFile] 1110332x3072 instances loaded from /root/VisualSearch/tvr/FeatureData/i3d_resnet
2023-06-22 07:59:31.275:INFO:main - model_config {'visual_input_size': 3072, 'query_input_size': 768, 'hidden_size': 384, 'max_ctx_l': 128, 'max_desc_l': 30, 'map_size': 32, 'input_drop': 0.1, 'device': [0], 'drop': 0.1, 'n_heads': 4, 'initializer_range': 0.02, 'margin': 0.1, 'use_hard_negative': False, 'hard_pool_size': 20}
Traceback (most recent call last):
File "method/train.py", line 264, in
model_dir, eval_split_name, eval_path, debug, model_name = start_training(opt)
File "method/train.py", line 251, in start_training
model = NAME_TO_MODELSopt.model_name
KeyError: 'MS-SL_Net'