Newer
Older
#!/bin/bash
export CIFAR10_PATH_BASE=<path_to_this_repo>
export CIFAR10_PYTHON_FILE_NAME=<cifar10_cnn_single_gpu.py or cifar10_cnn_multi_gpu.py>
export CIFAR10_NUM_GPUS=<1-4>
export CIFAR10_BATCHSIZE=<64-2048>
export CIFAR10_NUM_CORES=<1-16>
export CIFAR10_PATH_SAVED_MODELS=<path_to_saved_models_folder>
export CIFAR10_PATH_TENSORBOARD_LOGS=<path_to_tensorboard_logs_folder>
export KERAS_BACKEND=tensorflow