int main (int argc, char** argv) { cudaSetDevice(0); #ifdef linux signal(SIGSEGV, handler); // install our handler #endif srand(clock()); if(argc >= 3){ g_argv.push_back(atoi(argv[1])); g_argv.push_back(atoi(argv[2])); } printf("1. MNIST\n2. CIFAR-10\n3. CHINESE\n4. CIFAR-100\n5. VOTE MNIST\nChoose the dataSet to run:"); int cmd; if(g_argv.size() >= 2) cmd = g_argv[0]; else if(1 != scanf("%d", &cmd)){ LOG("scanf fail", "result/log.txt"); } if(cmd == 1) runMnist(); else if(cmd == 2) runCifar10(); else if(cmd == 3) runChinese(); else if(cmd == 4) runCifar100(); else if(cmd == 5) cuVoteMnist(); return EXIT_SUCCESS; }
int main (void) { runMnist(); //cuPredict(); return EXIT_SUCCESS; }