← back to Handbag Authentication

handbag_data/github_datasets/iGAN/train_dcgan/train_script.sh

8 lines

MODEL_NAME=$1
THEANO_FLAGS='device=gpu0, floatX=float32, nvcc.fastmath=True' python train_dcgan.py --model_name $MODEL_NAME
THEANO_FLAGS='device=gpu0, floatX=float32, nvcc.fastmath=True' python batchnorm_dcgan.py --model_name $MODEL_NAME
THEANO_FLAGS='device=gpu0, floatX=float32, nvcc.fastmath=True' python train_predict_z.py --model_name $MODEL_NAME
THEANO_FLAGS='device=gpu0, floatX=float32, nvcc.fastmath=True' python batchnorm_precit_z.py --model_name $MODEL_NAME
python pack_model.py --model_name $MODEL_NAME