#!/bin/bash

export TF_FORCE_GPU_ALLOW_GROWTH=true

cd /ProteinGAN/src/gan/

python -u -m train_gan --batch_size 16 --name test_run --steps 12000 -shuffle_buffer_size 100000 --loss_type non_saturating --discriminator_learning_rate 0.0001 --generator_learning_rate 0.0001 --dilation_rate 2 --gf_dim 44 --df_dim 30 --dataset protein/Length_512_lab_test_37_v2 --architecture gumbel --pooling conv
