Skip to content

Instantly share code, notes, and snippets.

@vabh

vabh/internship_readme.md

Last active July 4, 2016 11:45
Show Gist options
  • Save vabh/ea7e8032a045a052f6fb03952ae5681a to your computer and use it in GitHub Desktop.
Save vabh/ea7e8032a045a052f6fb03952ae5681a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
internship_readme.md

Train a model
caffe_root/build/tools/caffe train -solver [path_to_solver_file] [Trained models are saved in a location specified in the solver file]

accuracy_cifar.py - generates a file with (iterations, accuracy)
python accuracy_cifar.py [network_definition] [model_prefix] [gpu]
network_deifinition - the .prototxt file which has referenced in the solver. This contains the network which was trained.
model_prefix - this is the prefix with which models were saved throughout the training phase.

For both network_definition and model_prefix, the root folders are specified inside the script.

accuracy_plot.py - plots from input text files. The labels are to be specified inside the script, depending on mutiple files have multiple series or a single file has mult-columns, each being a different series.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment