This repo contains a modified ResNet (https://arxiv.org/abs/1512.03385) implementation and code to train it on Tiny Imagenet. The model and training code is implemented with PyTorch. Before running 'train.py' you need to execute 'download_tinyimagenet.sh' to download the dataset.
cschoeller / tinyimagenet Goto Github PK
View Code? Open in Web Editor NEWTiny Imagenet Classification with ResNet
License: MIT License