Cppowboy / wrn-tensorflow

A tensorflow implementation of Wide Residual Networks(https://arxiv.org/abs/1605.07146)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

wrn-tensorflow

A tensorflow implementation of Wide Residual Networks(https://arxiv.org/abs/1605.07146)

Reference:

  1. Original code: https://github.com/szagoruyko/wide-residual-networks
  2. The data module cifar100_input.py is from TensorFlow tutorial.

Prerequisite

  1. TensorFlow

How To Run

# Clone the repo.
git clone https://github.com/dalgu90/wrn-tensorflow.git
cd wrn-tensorflow

# Download CIFAR-100 dataset
wget https://www.cs.toronto.edu/~kriz/cifar-100-binary.tar.gz
tar xvf cifar-100-binary.tar.gz

# Run! (WRN-28-1)
./train.sh

# To evaluate
./eval.sh wrn_28_1 4 1

About

A tensorflow implementation of Wide Residual Networks(https://arxiv.org/abs/1605.07146)


Languages

Language:Python 97.9%Language:Shell 2.1%