加载预训练模型,报错Tensor' object has no attribute 'get
Traceback (most recent call last): File "", line 1, in File "D:\ide\PyCharm 2019.2.1\helpers\pydev_pydev_bundle\pydev_umd.py", line 197, in runfile pydev_imports.execfile(filename, global_vars, local_vars) # execute the script File "D:\ide\PyCharm 2019.2.1\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "D:/qiangdao/py_project/PaddleSpeech/audio/examples/sound_classification/export_model.py", line 42, in model.set_state_dict(paddle.load(args.checkpoint)) File "C:\Users\qiegewala\anaconda3\envs\rrr\lib\site-packages\paddle\fluid\framework.py", line 281, in wrapper return func(*args, **kwargs) File "C:\Users\qiegewala\anaconda3\envs\rrr\lib\site-packages\paddle\fluid\dygraph\layers.py", line 1435, in set_state_dict match_res = _check_match(key_name, param) File "C:\Users\qiegewala\anaconda3\envs\rrr\lib\site-packages\paddle\fluid\dygraph\layers.py", line 1410, in _check_match state = state_dict.get(key, None) AttributeError: 'Tensor' object has no attribute 'get'
model = SoundClassifier( backbone=cnn14(pretrained=False, extract_embedding=True), num_class=len(ESC50.label_list)) model.set_state_dict(paddle.load(args.checkpoint)) model.eval()
model = paddle.jit.to_static(
model,
input_spec=[
paddle.static.InputSpec(
shape=[None, None, 64], dtype=paddle.float32)
])
windows,paddlepaddle-gpu=2.2.0,python3.7