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 "
|
DATA=" ddpm.input_dim 3 data.cates car "
|
||||||
NGPU=$1 #
|
NGPU=$1 #
|
||||||
num_node=1
|
num_node=1
|
||||||
BS=32
|
BS=6
|
||||||
total_bs=$(( $NGPU * $BS ))
|
total_bs=$(( $NGPU * $BS ))
|
||||||
if (( $total_bs > 128 )); then
|
if (( $total_bs > 128 )); then
|
||||||
echo "[WARNING] total batch_size larger than 128 may lead to unstable training, please reduce the size"
|
echo "[WARNING] total batch_size larger than 128 may lead to unstable training, please reduce the size"
|
||||||
|
|
Loading…
Reference in a new issue