Error when loading pretrained VGG19

[ntomita]

I encountered an error when loading a pretrained vgg19 model. I tried other networks like Alexnet and they're seemed fine.

I use torch==0.1.10, torchvision==0.1.7, just updated from master branch. Also, it's under Python3.6 on latest macOS.

Do anyone have any idea how this happened and can be fixed? An error message seems like complaining incompatible un-pickling.

model = models.vgg19(True)
Traceback (most recent call last):
File "", line 1, in
File "/usr/local/lib/python3.6/site-packages/torchvision/models/vgg.py", line 141, in vgg19
model.load_state_dict(model_zoo.load_url(model_urls['vgg19']))
File "/usr/local/lib/python3.6/site-packages/torch/utils/model_zoo.py", line 57, in load_url
return torch.load(cached_file)
File "/usr/local/lib/python3.6/site-packages/torch/serialization.py", line 222, in load
return _load(f, map_location, pickle_module)
File "/usr/local/lib/python3.6/site-packages/torch/serialization.py", line 370, in _load
result = unpickler.load()
AttributeError: Can't get attribute '_rebuild_tensor' on <module 'torch._utils' from '/usr/local/lib/python3.6/site-packages/torch/_utils.py'>

[ntomita]

I also note that loading vgg11 and vgg13 is fine. vgg16 and vgg19 produces the error.

[soumith]

what version of torch are you on?

[ntomita]

I'm using torch==0.1.10, this is the newest one, right?

[soumith]

yes it is, I'll investigate and fix these checkpoints on Monday. thanks for letting me know.

[ntomita]

Thanks!

[soumith]

this will be fixed after updating to the pytorch release v0.1.11 on wednesday.

[ntomita]

Thank you very much for your quick response! I'm looking forward to it.

[Cadene]

I encouter the same error for vgg16.

CUDA_VISIBLE_DEVICES=2 python main.py /local/cadene/data --arch vgg16 -b 20 --lr 1e-4 --pretrained
=> using pre-trained model 'vgg16'
Downloading: "https://download.pytorch.org/models/vgg16-397923af.pth" to /home/cadene/.torch/models/vgg16-397923af.pth
100.0%
Traceback (most recent call last):
  File "main.py", line 335, in <module>
    main()
  File "main.py", line 120, in main
    model = models.__dict__[args.arch](pretrained=True)
  File "build/bdist.linux-x86_64/egg/torchvision/models/vgg.py", line 124, in vgg16
    model.load_state_dict(model_zoo.load_url(model_urls['vgg16']))
  File "/home/cadene/anaconda3/envs/default/lib/python2.7/site-packages/torch/utils/model_zoo.py", line 57, in load_url
    return torch.load(cached_file)
  File "/home/cadene/anaconda3/envs/default/lib/python2.7/site-packages/torch/serialization.py", line 222, in load
    return _load(f, map_location, pickle_module)
  File "/home/cadene/anaconda3/envs/default/lib/python2.7/site-packages/torch/serialization.py", line 370, in _load
    result = unpickler.load()
AttributeError: 'module' object has no attribute '_rebuild_tensor'

[lanpa]

@Cadene I solved the same problem by installing newest pytorch and torchvision from source.

[alykhantejani]

@Cadene is this still an issue for you when using the latest pytorch?

[alykhantejani]

This appears to work with the latest version of PyTorch so closing this issue. Let me know if you are still having issues.

[varghesealex90]

Getting an error when trying to import inception model. Any fixes. Same with densenet. However all models of resnet loading without any issues

[varghesealex90]

inception = torchvision.models.inception_v3()
Traceback (most recent call last):
File "", line 1, in
AttributeError: 'module' object has no attribute 'inception_v3'