xkuang/CNN-Tutorial

Convolutional Neural Networks Tutorial Materials, Jan 2018

Convolutional Neural Networks Tutorial

Women in Machine Learning and Data Science, LinkedIn, 18 January 2018

Usage

Installation

1. Install Anaconda or Miniconda
2. Run the following commands

conda env create -n cnn_tutorial -f environment.yml
source activate cnn_tutorial

Create data and model folders

mkdir data
mkdir models

To train a basic CNN with default settings

python main.py

To train a VGG like net

python main.py --nettype 1

To train a Residual Network

python main.py --nettype 2

Results after 20 epochs:

1. Basic CNN:

• Train:
• Test:

2. VGG like net

• Train:
• Test:

3. ResNet

• Train:
• Test:

Sources

• Pytorch CIFAR10 tutorial
• VGG paper
• ResNet paper