This repository is to practice my PyTorch for image classficiation problem.
The dataset is from https://www.cs.toronto.edu/~kriz/cifar.html
Downloaded data needs to be stored in ./data
folder
Files to execute:
EDA.ipynb
: explore the raw datapreprocess.ipynb
: preprocess the raw datasanity_checks.py
: sanity checks for the implementationhyper_tune.py
: tune hyperparameterstrain.py
: train the model based on best hyperparameters searched in step 4
The implemeted model here is based on the ResNet paper: Deep Residual Learning for Image Recognition [He et al. 2015]