Fix location of dropout in VGG

[colesbury]

Fixes #92

I've re-trained the VGG models on ImageNet. The fixed dropout location improved the accuracy.

[colesbury]

For posterity:

I trained the models using the ImageNet example:

python3 -u main.py -a vgg11 -j 16 --lr 0.01 /path/to/imagenet/ | tee vgg11.stdout.log
python3 -u main.py -a vgg13 -j 16 --lr 0.01 /path/to/imagenet/ | tee vgg13.stdout.log
python3 -u main.py -a vgg16 -j 16 --lr 0.01 /path/to/imagenet/ | tee vgg16.stdout.log
python3 -u main.py -a vgg19 -j 16 --lr 0.01 /path/to/imagenet/ | tee vgg19.stdout.log

vgg11.stdout.log
vgg13.stdout.log
vgg16.stdout.log
vgg19.stdout.log

[ecolss]

@colesbury

Downloaded this latest vgg19 model, but error occurs when loading it:
th.load('/home/.torch/models/vgg19-dcbb9e9d.pth')

/usr/local/lib/python3.6/site-packages/torch/serialization.py in load(f, map_location, pickle_module)
    220         f = open(f, 'rb')
    221     try:
--> 222         return _load(f, map_location, pickle_module)
    223     finally:
    224         if new_fd:

/usr/local/lib/python3.6/site-packages/torch/serialization.py in _load(f, map_location, pickle_module)
    368     unpickler = pickle_module.Unpickler(f)
    369     unpickler.persistent_load = persistent_load
--> 370     result = unpickler.load()
    371
    372     deserialized_storage_keys = pickle_module.load(f)

AttributeError: Can't get attribute '_rebuild_tensor' on <module 'torch._utils' from '/usr/local/lib/python3.6/site-packages/torch/_utils.py'>

[colesbury]

You need to be running the latest version of PyTorch as well. You may have to build from master for now.

…

On Mon, Mar 20, 2017 at 9:03 AM ecolss ***@***.***> wrote: @colesbury <https://github.com/colesbury> Downloaded this latest vgg19 model, but error occurs when loading it: /usr/local/lib/python3.6/site-packages/torch/serialization.py in load(f, map_location, pickle_module) 220 f = open(f, 'rb') 221 try: --> 222 return _load(f, map_location, pickle_module) 223 finally: 224 if new_fd: /usr/local/lib/python3.6/site-packages/torch/serialization.py in _load(f, map_location, pickle_module) 368 unpickler = pickle_module.Unpickler(f) 369 unpickler.persistent_load = persistent_load --> 370 result = unpickler.load() 371 372 deserialized_storage_keys = pickle_module.load(f) AttributeError: Can't get attribute '_rebuild_tensor' on <module 'torch._utils' from '/usr/local/lib/python3.6/site-packages/torch/_utils.py'> — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub <#107 (comment)>, or mute the thread <https://github.com/notifications/unsubscribe-auth/AAoB-hM_hKO1YqmJtnx0lwtLflpZifs8ks5rnnkggaJpZM4MfWO3> .

[soumith]

this error would be fixed with the release on Wednesday v0.1.11.

[ecolss]

I see, thanks