Safran/LION
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: change batch size to fit on bigviz

Browse source
This commit is contained in:
Laurent FAINSIN 2023-04-07 13:32:44 +02:00
parent cb3c32d6b1
commit 7225eabb15
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
script/train_vae.sh Normal file → Executable file
View file

@ -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"