From 5264f262b0163868926754cedd5b5e5e9cc240b0 Mon Sep 17 00:00:00 2001
From: Jean-Luc Parouty <uja62cb@jean-zay4.ib0.xa.idris.fr>
Date: Wed, 6 Jan 2021 22:41:54 +0100
Subject: [PATCH] update VAE batch script

---
 VAE/batch_slurm.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/VAE/batch_slurm.sh b/VAE/batch_slurm.sh
index ff1474d..503885f 100755
--- a/VAE/batch_slurm.sh
+++ b/VAE/batch_slurm.sh
@@ -46,7 +46,8 @@ export FIDLE_OVERRIDE_VAE8_run_dir="./run/CelebA.$SLURM_JOB_ID"
 export FIDLE_OVERRIDE_VAE8_scale="1"
 export FIDLE_OVERRIDE_VAE8_image_size="(128,128)"
 export FIDLE_OVERRIDE_VAE8_enhanced_dir='{datasets_dir}/celeba/enhanced'
-export FIDLE_OVERRIDE_VAZ8_r_loss_factor="0.7"
+export FIDLE_OVERRIDE_VAE8_r_loss_factor="0.7"
+export FIDLE_OVERRIDE_VAE8_epochs="8"
 
 # ---- By default (no need to modify)
 #
-- 
GitLab