Ejemplo n.º 1
0
def run():
	global USE_GPUS
	val_num = np.random.randint(1, 6)
	cifar10_input.set_constants(train=True, val_num=val_num)
	if USE_GPUS:			
		import cifar10_multi_gpu_train
		cifar10_multi_gpu_train.train()
	else:
		import cifar10_train
		cifar10_train.main()
	cifar10_input.set_constants(train=False, val_num=val_num)
	cifar10_eval.main()
Ejemplo n.º 2
0
def run():
	global USE_GPUS
	# val_num = np.random.randint(1, 6)
	# cifar10_input.set_constants(train=True, val_num=val_num)
	# if USE_GPUS:			
	# 	import cifar10_multi_gpu_train
	# 	cifar10_multi_gpu_train.main()
	# else:
	# 	import cifar10_train
	# 	cifar10_train.main()
	cifar10_input.set_constants(train=False, val_num=1)
	cifar10_eval.main()