From 7225eabb151c81e43600bee0a37d0257ed34579b Mon Sep 17 00:00:00 2001
From: Laurent FAINSIN <laurent.fainsin@safrangroup.com>
Date: Fri, 7 Apr 2023 13:32:44 +0200
Subject: [PATCH] fix: change batch size to fit on bigviz

---
 script/train_vae.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
 mode change 100644 => 100755 script/train_vae.sh

diff --git a/script/train_vae.sh b/script/train_vae.sh
old mode 100644
new mode 100755
index f5627e0..fe4468e
--- a/script/train_vae.sh
+++ b/script/train_vae.sh
@@ -6,7 +6,7 @@ fi
 DATA=" ddpm.input_dim 3 data.cates car "
 NGPU=$1 # 
 num_node=1
-BS=32 
+BS=6
 total_bs=$(( $NGPU * $BS ))
 if (( $total_bs > 128 )); then 
     echo "[WARNING] total batch_size larger than 128 may lead to unstable training, please reduce the size"