Traceback (most recent call last):
File "main.py", line 42, in
main()
File "main.py", line 37, in main
solver = Solver(module, opt)
File "/home/weiyu/SelfReformer/solver.py", line 18, in init
self.net = module.Net(opt)
File "/home/weiyu/SelfReformer/model/network.py", line 14, in init
self.encoder = Encoder()
File "/home/weiyu/SelfReformer/model/Encoder_pvt.py", line 13, in init
self.encoder.load_state_dict(torch.load('./model/pretrain/pvt_v2_b2.pth', map_location='cpu'), strict=False)
File "/home/weiyu/anaconda3/envs/selfReformer/lib/python3.8/site-packages/torch/serialization.py", line 593, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/weiyu/anaconda3/envs/selfReformer/lib/python3.8/site-packages/torch/serialization.py", line 762, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.