BeatCraft / LDNN-mnist

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LDNN-mnist

Install

clone or unzip the archive of this repository
cd LDNN-mnist
mkdir data
mkdir batch
mkdir wi

Download the dataset

THE MNIST DATABASE of handwritten digits
http://yann.lecun.com/exdb/mnist/

put these 4 files to batch

  • t10k-images-idx3-ubyte
  • t10k-labels-idx1-ubyte
  • train-images-idx3-ubyte
  • train-labels-idx1-ubyte

Prepare

  1. check path fot LesserDNN
    in .py

sys.path.append(os.path.join(os.path.dirname(file), '../ldnn'))

this line includes LesserDNN framework:)
check if "../ldnn" is correct on your env.

  1. create batch files for training and test
    python3 ./make_batch

  2. test
    ./test.sh

Training

./train.sh

Select opencl / cupy

modiy "../ldnn/plat.py"

About


Languages

Language:Python 90.0%Language:Shell 10.0%