-
Notifications
You must be signed in to change notification settings - Fork 77
Description
Traceback (most recent call last):
File "D:\pytest\AOT-GAN-for-Inpainting-master\src\demo.py", line 112, in
demo(args)
File "D:\pytest\AOT-GAN-for-Inpainting-master\src\demo.py", line 32, in demo
model.load_state_dict(torch.load(args.pre_train, map_location="cpu")) #加载预训练模型参数,map_location="cpu" 表示即使没有 GPU 也能运行
File "D:\anaconda\anaconda3\envs\inpainting\lib\site-packages\torch\nn\modules\module.py", line 1223, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for InpaintGenerator:
Missing key(s) in state_dict: "encoder.1.weight", "encoder.1.bias", "encoder.3.weight", "encoder.3.bias", "encoder.5.weight", "encoder.5.bias", "middle.0.block00.1.weight", "middle.0.block00.1.bias", "middle.0.block01.1.weight", "middle.0.block01.1.bias", "middle.0.block02.1.weight", "middle.0.block02.1.bias", "middle.0.block03.1.weight", "middle.0.block03.1.bias", "middle.0.fuse.1.weight", "middle.0.fuse.1.bias", "middle.0.gate.1.weight", "middle.0.gate.1.bias", "middle.1.block00.1.weight", "middle.1.block00.1.bias", "middle.1.block01.1.weight", "middle.1.block01.1.bias", "middle.1.block02.1.weight", "middle.1.block02.1.bias", "middle.1.block03.1.weight", "middle.1.block03.1.bias", "middle.1.fuse.1.weight", "middle.1.fuse.1.bias", "middle.1.gate.1.weight", "middle.1.gate.1.bias", "middle.2.block00.1.weight", "middle.2.block00.1.bias", "middle.2.block01.1.weight", "middle.2.block01.1.bias", "middle.2.block02.1.weight", "middle.2.block02.1.bias", "middle.2.block03.1.weight", "middle.2.block03.1.bias", "middle.2.fuse.1.weight", "middle.2.fuse.1.bias", "middle.2.gate.1.weight", "middle.2.gate.1.bias", "middle.3.block00.1.weight", "middle.3.block00.1.bias", "middle.3.block01.1.weight", "middle.3.block01.1.bias", "middle.3.block02.1.weight", "middle.3.block02.1.bias", "middle.3.block03.1.weight", "middle.3.block03.1.bias", "middle.3.fuse.1.weight", "middle.3.fuse.1.bias", "middle.3.gate.1.weight", "middle.3.gate.1.bias", "middle.4.block00.1.weight", "middle.4.block00.1.bias", "middle.4.block01.1.weight", "middle.4.block01.1.bias", "middle.4.block02.1.weight", "middle.4.block02.1.bias", "middle.4.block03.1.weight", "middle.4.block03.1.bias", "middle.4.fuse.1.weight", "middle.4.fuse.1.bias", "middle.4.gate.1.weight", "middle.4.gate.1.bias", "middle.5.block00.1.weight", "middle.5.block00.1.bias", "middle.5.block01.1.weight", "middle.5.block01.1.bias", "middle.5.block02.1.weight", "middle.5.block02.1.bias", "middle.5.block03.1.weight", "middle.5.block03.1.bias", "middle.5.fuse.1.weight", "middle.5.fuse.1.bias", "middle.5.gate.1.weight", "middle.5.gate.1.bias", "middle.6.block00.1.weight", "middle.6.block00.1.bias", "middle.6.block01.1.weight", "middle.6.block01.1.bias", "middle.6.block02.1.weight", "middle.6.block02.1.bias", "middle.6.block03.1.weight", "middle.6.block03.1.bias", "middle.6.fuse.1.weight", "middle.6.fuse.1.bias", "middle.6.gate.1.weight", "middle.6.gate.1.bias", "middle.7.block00.1.weight", "middle.7.block00.1.bias", "middle.7.block01.1.weight", "middle.7.block01.1.bias", "middle.7.block02.1.weight", "middle.7.block02.1.bias", "middle.7.block03.1.weight", "middle.7.block03.1.bias", "middle.7.fuse.1.weight", "middle.7.fuse.1.bias", "middle.7.gate.1.weight", "middle.7.gate.1.bias", "decoder.0.conv.weight", "decoder.0.conv.bias", "decoder.2.conv.weight", "decoder.2.conv.bias", "decoder.4.weight", "decoder.4.bias".
Unexpected key(s) in state_dict: "conv.0.weight_orig", "conv.0.weight_u", "conv.0.weight_v", "conv.2.weight_orig", "conv.2.weight_u", "conv.2.weight_v", "conv.4.weight_orig", "conv.4.weight_u", "conv.4.weight_v", "conv.6.weight_orig", "conv.6.weight_u", "conv.6.weight_v", "conv.8.weight", "conv.8.bias".