/deep

neural networks, using caffe, etc.

Primary LanguageJupyter Notebook

pycaffe

$ ls caffe/ # python caffe wrapper. This folder lets us import caffe if we want to models/ # stores model snapshots (.caffemodel, .solverstate) nets/ # Network definitions (.prototxt)

Datasets are stored on /scratch/rshaban1/lmdb/.

Things you might run

python generate_prototxt.py > output.prototxt

caffe.bin train -solver net_solver.prototxt

caffe.bin test -solver solver.prototxt -weights iter_4000.caffemodel.h5 -model train_test.prototxt

python /scratch/rshaban1/caffe/python/draw_net.py nets/cifar_cnn/5_3_2x_cnn.prototxt nets/cifar_cnn/5_3_2x.png