配置好框架之后 train.py训练 报这个错 请问大家 我是哪里还没配好吗
File "train.py", line 213, in
main(args)
File "train.py", line 164, in main
model = build_model(cfg.model)
File "/home/liyanyan/PSE_origial/models/builder.py", line 11, in build_model
model = models.dictcfg.type
File "/home/liyanyan/PSE_origial/models/psenet.py", line 20, in init
self.backbone = build_backbone(backbone)
File "/home/liyanyan/PSE_origial/models/backbone/builder.py", line 11, in build_backbone
backbone = models.backbone.dictcfg.type
File "/home/liyanyan/PSE_origial/models/backbone/resnet.py", line 210, in resnet50
model.load_state_dict(load_url(model_urls['resnet50']), strict=False)
File "/home/liyanyan/PSE_origial/models/backbone/resnet.py", line 234, in load_url
return torch.load(cached_file, map_location=map_location)
File "/home/liyanyan/.conda/envs/PSE-env/lib/python3.7/site-packages/torch/serialization.py", line 387, in load
return _load(f, map_location, pickle_module, pickle_load_args)
File "/home/liyanyan/.conda/envs/PSE-env/lib/python3.7/site-packages/torch/serialization.py", line 564, in _load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, '\xef'.
配置好框架之后 train.py训练 报这个错 请问大家 我是哪里还没配好吗 File "train.py", line 213, in
main(args)
File "train.py", line 164, in main
model = build_model(cfg.model)
File "/home/liyanyan/PSE_origial/models/builder.py", line 11, in build_model
model = models.dictcfg.type
File "/home/liyanyan/PSE_origial/models/psenet.py", line 20, in init
self.backbone = build_backbone(backbone)
File "/home/liyanyan/PSE_origial/models/backbone/builder.py", line 11, in build_backbone
backbone = models.backbone.dictcfg.type
File "/home/liyanyan/PSE_origial/models/backbone/resnet.py", line 210, in resnet50
model.load_state_dict(load_url(model_urls['resnet50']), strict=False)
File "/home/liyanyan/PSE_origial/models/backbone/resnet.py", line 234, in load_url
return torch.load(cached_file, map_location=map_location)
File "/home/liyanyan/.conda/envs/PSE-env/lib/python3.7/site-packages/torch/serialization.py", line 387, in load
return _load(f, map_location, pickle_module, pickle_load_args)
File "/home/liyanyan/.conda/envs/PSE-env/lib/python3.7/site-packages/torch/serialization.py", line 564, in _load
magic_number = pickle_module.load(f, pickle_load_args)
_pickle.UnpicklingError: invalid load key, '\xef'.