vgg16_features.load_state_dict(torch.load(vgg_path), strict=False) about fast-neural-style-pytorch HOT 4 CLOSED

On Mon, Feb 3, 2020, 10:02 PM Fabio Vianello ***@***.***> wrote: I have no issues with stylize.py (with pretrained models). My problem is with train.py. I think I followed the instructions: downloaded vgg16-00b39a1b.pth, downloaded train2014, installed pytorch, opencv, etc (installed with conda). I have Debian 11 stable, Python 3.7.4, pytorch 1.4.0, torchvision 0.5.0, cudatoolkit 10.1.243. $ python train.py Traceback (most recent call last): File "train.py", line 168, in <module> train() File "train.py", line 50, in train VGG = vgg.VGG16().to(device) File "/home/mrfabiolo/style_transfer/fast-neural-style-pytorch/vgg.py", line 33, in __init__ vgg16_features.load_state_dict(torch.load(vgg_path), strict=False) File "/home/mrfabiolo/miniconda3/lib/python3.7/site-packages/torch/serialization.py", line 529, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/home/mrfabiolo/miniconda3/lib/python3.7/site-packages/torch/serialization.py", line 692, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'. Same problem with device = "cpu" or device = "cuda". I got a GeForce GTX 750 card. — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#10?email_source=notifications&email_token=AB3NRXJOWZTPFVKQQICZ3NDRBAIX3A5CNFSM4KPEZMEKYY3PNVWWK3TUL52HS4DFUVEXG43VMWVGG33NNVSW45C7NFSM4IKSXDVA>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AB3NRXJACJFRHWE3VK47GW3RBAIX3ANCNFSM4KPEZMEA> .