uoguelph-mlrg / Cutout

2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552
Other
542 stars 156 forks source link

Type uint8_t overflow fix and python3, pytorch 0.4.0 compatibility #1

Closed raeidsaqur closed 6 years ago

raeidsaqur commented 6 years ago

Minor changes: please review and merge if you deem fit. Tested with pytorch 0.4.0 and python3

TDeVries commented 6 years ago

Thank you for updating to PyTorch 0.4.0 and Python3!