PyTorch implementation of SNAPSHOT ENSEMBLES: TRAIN 1, GET M FOR FREE

Corrected some errors in the code by moskomule and adapted the code to the latest PyTorch (version 1.10).

Several modifications have also been made to improve the performance in test.ipynb.

basic.py is simple network for CIFAR10 classification.