From 84b33c27dc9feaf94bb33b8fabfee4f7d6a1570c Mon Sep 17 00:00:00 2001 From: Guandao Yang Date: Mon, 15 Jul 2019 20:50:28 -0700 Subject: [PATCH] Update: batch size. Reduce the batch size for AE, so that it fits into single GPU. --- scripts/shapenet_airplane_ae.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/shapenet_airplane_ae.sh b/scripts/shapenet_airplane_ae.sh index 6768dbe..6f8d052 100755 --- a/scripts/shapenet_airplane_ae.sh +++ b/scripts/shapenet_airplane_ae.sh @@ -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