From 3cf2819c73b713be1d098b6671e4163ae3e4c0a1 Mon Sep 17 00:00:00 2001
From: Jean-Luc Parouty <Jean-Luc.Parouty@grenoble-inp.fr>
Date: Thu, 10 Mar 2022 13:37:40 +0100
Subject: [PATCH] Update ci for gans

---
 fidle/batch-tests-slurm.sh |  2 +-
 fidle/ci/smart_gpu.yml     | 18 ++++++++++++++++--
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/fidle/batch-tests-slurm.sh b/fidle/batch-tests-slurm.sh
index 5d57577..c01aeeb 100644
--- a/fidle/batch-tests-slurm.sh
+++ b/fidle/batch-tests-slurm.sh
@@ -32,7 +32,7 @@ NOTEBOOK_DIR="$WORK/fidle/fidle"
 
 export FIDLE_OVERRIDE_RUNCI_profile_name='./ci/smart_gpu.yml'
 export FIDLE_OVERRIDE_RUNCI_reset='False'
-export FIDLE_OVERRIDE_RUNCI_filter='Nb_VAE9_r.*'
+export FIDLE_OVERRIDE_RUNCI_filter='Nb_SHEEP.*'
 
 NOTEBOOK_SRC1="02-running-ci-tests.ipynb"
 
diff --git a/fidle/ci/smart_gpu.yml b/fidle/ci/smart_gpu.yml
index e65f551..395a1bd 100644
--- a/fidle/ci/smart_gpu.yml
+++ b/fidle/ci/smart_gpu.yml
@@ -535,12 +535,26 @@ Nb_SHEEP1:
   notebook_tag: default
   overrides:
     scale: 1
-    run_dir: default
+    run_dir: ./run/SHEEP1
     latent_dim: default
-    epochs: 5
+    epochs: 10
     batch_size: 32
     num_img: 12
     fit_verbosity: 2
+
+Nb_SHEEP2:
+  notebook_id: SHEEP2
+  notebook_dir: DCGAN
+  notebook_src: 02-WGANGP-Draw-me-a-sheep.ipynb
+  notebook_tag: default
+  overrides:
+    scale: 1
+    run_dir: ./run/SHEEP2
+    latent_dim: 80
+    epochs: 3
+    batch_size: 64
+    num_img: 12
+    fit_verbosity: 2    
 #
 # ------ Misc ------------------------------------------------------
 #
-- 
GitLab