Code for the Kaggle CDiscount Image Classification Challenge.
Structure:
examples
: pytorch examplesmongo_helper.py
: Helper class to interact with mongodb, where the data is storeddataset.py
: Class used by pytorch to load datanet.py
: Script to train and test model.