The NeXtVLAD final model couldn't be used ???
Created by: ucas010
hi,dear the nextvlad_final file can't be load ???
[INFO: predict.py: 185]: Namespace(batch_size=1, config='configs/attention_cluster.txt', filelist=None, infer_topk=20, log_interval=1, model_name='AttentionCluster', save_dir='data\\predict_results', use_gpu=False, video_path=None, weights='./NEXTVLAD_final.pdparams')
[INFO: config_utils.py: 56]: ---------------- Infer Arguments ----------------
[INFO: config_utils.py: 58]: MODEL:
[INFO: config_utils.py: 60]: name:AttentionCluster
[INFO: config_utils.py: 60]: dataset:YouTube-8M
[INFO: config_utils.py: 60]: bone_network:None
[INFO: config_utils.py: 60]: drop_rate:0.5
[INFO: config_utils.py: 60]: feature_num:2
[INFO: config_utils.py: 60]: feature_names:['rgb', 'audio']
[INFO: config_utils.py: 60]: feature_dims:[1024, 128]
[INFO: config_utils.py: 60]: seg_num:100
[INFO: config_utils.py: 60]: cluster_nums:[32, 32]
[INFO: config_utils.py: 60]: num_classes:3862
[INFO: config_utils.py: 60]: topk:20
[INFO: config_utils.py: 60]: UNIQUE:{'good': 20, 'bad': 30}
[INFO: config_utils.py: 58]: TRAIN:
[INFO: config_utils.py: 60]: epoch:5
[INFO: config_utils.py: 60]: learning_rate:0.001
[INFO: config_utils.py: 60]: pretrain_base:None
[INFO: config_utils.py: 60]: batch_size:2048
[INFO: config_utils.py: 60]: use_gpu:True
[INFO: config_utils.py: 60]: num_gpus:8
[INFO: config_utils.py: 60]: filelist:data/dataset/youtube8m/train.list
[INFO: config_utils.py: 58]: VALID:
[INFO: config_utils.py: 60]: batch_size:2048
[INFO: config_utils.py: 60]: filelist:data/dataset/youtube8m/val.list
[INFO: config_utils.py: 58]: TEST:
[INFO: config_utils.py: 60]: batch_size:256
[INFO: config_utils.py: 60]: filelist:data/dataset/youtube8m/test.list
[INFO: config_utils.py: 58]: INFER:
[INFO: config_utils.py: 60]: batch_size:1
[INFO: config_utils.py: 60]: filelist:data/dataset/youtube8m/infer.list
[INFO: config_utils.py: 61]: -------------------------------------------------
Traceback (most recent call last):
File "D:\python36\new\rcdnn\image\predict.py", line 187, in <module>
infer(args)
File "D:\python36\new\rcdnn\image\predict.py", line 90, in infer
infer_model = models.get_model(args.model_name, infer_config, mode='infer')
File "D:\python36\new\rcdnn\image\models\model.py", line 191, in get_model
return model_zoo.get(name, cfg, mode)
File "D:\python36\new\rcdnn\image\models\model.py", line 179, in get
raise ModelNotFoundError(name, self.model_zoo.keys())
models.model.ModelNotFoundError: Model AttentionCluster Not Found.
Availiable models:
NEXTVLAD
Except specified the final model path, no other modified. SO how to solve the problem? thx