szagoruyko / attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)
https://arxiv.org/abs/1612.03928
1.44k stars 276 forks source link

Sync with torchvision #6

Closed szagoruyko closed 7 years ago

szagoruyko commented 7 years ago

Fixes error in DataLoader cause by transpose in internal representation of CIFAR10 dataset in torchvision + pep8ing