From f943f7e920e8b73384c2ad9610110b855632b42c Mon Sep 17 00:00:00 2001 From: muhanzhang Date: Wed, 4 Sep 2019 22:11:50 -0500 Subject: [PATCH] change default batch size to 1 --- run_DGCNN.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_DGCNN.sh b/run_DGCNN.sh index 58dd349..f38fc75 100755 --- a/run_DGCNN.sh +++ b/run_DGCNN.sh @@ -13,7 +13,7 @@ CONV_SIZE="32-32-32-1" sortpooling_k=0.6 # If k <= 1, then k is set to an integer so that k% of graphs have nodes less than this integer FP_LEN=0 # final dense layer's input dimension, decided by data n_hidden=128 # final dense layer's hidden size -bsize=50 # batch size +bsize=1 # batch size, set to 50 or 100 to accelerate training dropout=True # dataset-specific settings