szagoruyko / attention-transfer

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

ImageNet PyTorch 0.4 #29

Closed szagoruyko closed 6 years ago

szagoruyko commented 6 years ago

Implemented and checked that it works. ResNet-18 with AT did not match numbers in the papers exactly, still looking.