Implementation of ResNet architecture in Pytorch
This network will train Resnet-50 architecture from scratch on CIFAR-10 dataset . If your device support cuda, it will be used automatically. The data will be downloaded and saved during the first run.
python CIFAR-10-train.py
All contributions are welcome. If you want to update some depricated piece of code or improve the existing code, feel free to send a pull request.