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 latest PyTorch + improve memory usage #4

Closed szagoruyko closed 7 years ago

szagoruyko commented 7 years ago