Skip to content
config.sh 409 B
Newer Older
Agnes Ansari's avatar
Agnes Ansari committed
#!/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