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
|
||||
latent_num_blocks=1
|
||||
zdim=128
|
||||
batch_size=48
|
||||
batch_size=16
|
||||
lr=2e-3
|
||||
epochs=4000
|
||||
ds=shapenet15k
|
||||
|
|
Loading…
Reference in a new issue