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

Update: batch size.

Browse source 
Reduce the batch size for AE, so that it fits into single GPU.
This commit is contained in:
Guandao Yang 2019-07-15 20:50:28 -07:00 committed by GitHub
parent 4e6795a4d9
commit 84b33c27dc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/shapenet_airplane_ae.sh
View file

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