Update: batch size.
Reduce the batch size for AE, so that it fits into single GPU.
This commit is contained in:
parent
4e6795a4d9
commit
84b33c27dc
|
@ -6,7 +6,7 @@ latent_dims="256-256"
|
||||||
num_blocks=1
|
num_blocks=1
|
||||||
latent_num_blocks=1
|
latent_num_blocks=1
|
||||||
zdim=128
|
zdim=128
|
||||||
batch_size=48
|
batch_size=16
|
||||||
lr=2e-3
|
lr=2e-3
|
||||||
epochs=4000
|
epochs=4000
|
||||||
ds=shapenet15k
|
ds=shapenet15k
|
||||||
|
|
Loading…
Reference in a new issue