search

sniklaus / pytorch-spynet

a reimplementation of Optical Flow Estimation using a Spatial Pyramid Network in PyTorch
GNU General Public License v3.0
317 stars 47 forks source link

Hello! How to solve this problem？ #10

Closed SaltwaterLHL closed 5 years ago

SaltwaterLHL commented 5 years ago

Traceback (most recent call last): File "/home/lhl/pytorch-spynet-master/run.py", line 131, in moduleNetwork = Network().cuda().eval() File "/home/lhl/pytorch-spynet-master/run.py", line 100, in init self.load_state_dict(torch.load('./network-' + arguments_strModel + '.pytorch')) File "/home/public/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 358, in load return _load(f, map_location, pickle_module) File "/home/public/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 532, in _load magic_number = pickle_module.load(f) _pickle.UnpicklingError: invalid load key, '<'.

sniklaus commented 5 years ago

Looks like downloading the weights again might solve this problem: https://github.com/sniklaus/pytorch-sepconv/issues/26#issuecomment-441220029