Skip to content

Latest commit

 

History

History
8 lines (5 loc) · 417 Bytes

README.md

File metadata and controls

8 lines (5 loc) · 417 Bytes

simple_cnn-pytorch

Human Protein Atlas Image Classification

Simple CNN written in PyTorch for the Human Protein Atlas Image Classification Kaggle challenge.

The data is available at https://www.kaggle.com/c/human-protein-atlas-image-classification/data after signing up with kaggle.

To run training, change the data_dir path in the train.py config parameters and run the train.py with no command line arguments.