Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
79 lines (58 sloc) 3.16 KB
// Please choose a data type to compile
#define DATATYPE 0
#include "marvin.hpp"
using namespace marvin;
using namespace std;
int main(int argc, char **argv){
if (argc < 3 || argc >10){
cout<<"Usage:"<<endl;
cout<<argv[0]<<" train network.json [model1.marvin[,model2.marvin,...]] [snapshot_iteration]"<<endl;
cout<<" example: "<<argv[0]<<" train examples/mnist/lenet.json"<<endl;
cout<<argv[0]<<" test network.json model1.marvin[,model2.marvin,...] response_name1[,name2,...] file_name1.tensor[,name2.tensor,...] [save_every_n_iterations]"<<endl;
cout<<" example: "<<argv[0]<<" test examples/mnist/lenet.json examples/mnist/lenet.marvin ip1,conv2 examples/mnist/ip1.tensor,examples/mnist/conv2.tensor"<<endl;
cout<<argv[0]<<" activate network.json model1.marvin[,model2.marvin,...] response_name_data response_name1[,name2,...] response1_channels[,response2_channels,...] file_prefix topK maxIterations"<<endl;
cout<<" example: "<<argv[0]<<" activate examples/mnist/lenet.json examples/mnist/lenet.marvin data conv1,conv2 [0,1,2],[0,1,2,3,4,5] examples/mnist/filters_ 100 20"<<endl;
return 0;
}
cout<< "====================================================================================================================================="<<endl;
cout<< ">> Hello, World! This is Marvin. I am at a rough estimate thirty billion times more intelligent than you. Let me give you an example."<<endl;
cout<< "====================================================================================================================================="<<endl;
if(0==strcmp(argv[1], "train")){
Solver solver(argv[2]);
solver.Malloc(Training);
solver.randInit();
if (argc==3){
solver.train();
}else if (argc==4 || argc==5){
vector<string> models = getStringVector(argv[3]);
for (int m=0;m<models.size();++m) solver.loadWeights(models[m],true);
if (argc==4){
solver.train();
}else{
solver.train(atoi(argv[4]));
}
}else FatalError(__LINE__);
solver.saveWeights(solver.path + ".marvin");
}else if(0==strcmp(argv[1], "test")){
Net net(argv[2]);
net.Malloc(Testing);
vector<string> models = getStringVector(argv[3]);
for (int m=0;m<models.size();++m) net.loadWeights(models[m]);
if (argc>=6){
int itersPerSave = 0;
if (argc==7){
itersPerSave = atoi(argv[6]);
}
net.test(getStringVector(argv[4]), getStringVector(argv[5]), itersPerSave);
}else if (argc==4){
net.test();
}else FatalError(__LINE__);
}else if(0==strcmp(argv[1], "activate")){
Net net(argv[2]);
net.Malloc(Testing);
vector<string> models = getStringVector(argv[3]);
for (int m=0;m<models.size();++m) net.loadWeights(models[m]);
net.getTopActivations(argv[4], getStringVector(argv[5]), getIntVectorVector(argv[6]), argv[7], atoi(argv[8]), atoi(argv[9]));
}
return 0;
}