Update CIFAR code to work with PyTorch 0.4

szagoruyko / attention-transfer

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

https://arxiv.org/abs/1612.03928

1.43k stars 274 forks source link

Closed szagoruyko closed 6 years ago

szagoruyko commented 6 years ago

This needs the laters torchvision and pytorch 0.4, I removed opencv dependency.

Going to update ImageNet code in separate PR.