fix: change batch size to fit on bigviz
This commit is contained in:
parent
cb3c32d6b1
commit
7225eabb15
2
script/train_vae.sh
Normal file → Executable file
2
script/train_vae.sh
Normal file → Executable 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"
|
||||
|
|
Loading…
Reference in a new issue