C++ (Cpp) runMnist Beispiele

Beispiel #1

Datei: main.cpp Projekt: Donkey-Sand/CUDA-CNN

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;
}

Beispiel #2

Datei: main.cpp Projekt: afib/CUDA-CNN

int main (void)
{
	runMnist();
	//cuPredict();
	return EXIT_SUCCESS;
}