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()
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()