remove_folders() {
	echo "Removing previous result directories..."
	# Reset train directory
	rm -rf ${TRAIN_DIR}
	mkdir -p ${TRAIN_DIR}
}


run_train() {                                                                  
	python ${SRC_AI_MODEL}/train_gan.py  \
		--codings_size       ${CODINGS_SIZE} \
		--seed        	     ${SEED} \
		--batch_size         ${BATCH_SIZE} \