aboutsummaryrefslogtreecommitdiff
path: root/config
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-07-25 14:19:40 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-07-25 14:19:40 -0400
commita557b939eb104ec7e0df42193e014e3137eb70f8 (patch)
treebbff025fb98dcd8d992bbdbc590fbd32f7227803 /config
parent5e216e8c5197568c39029793ebafc4df50215cae (diff)
downloadtaxi-a557b939eb104ec7e0df42193e014e3137eb70f8.tar.gz
taxi-a557b939eb104ec7e0df42193e014e3137eb70f8.zip
Change batch size for smaller GPU
Diffstat (limited to 'config')
-rw-r--r--config/bidirectional_tgtcls_2.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config/bidirectional_tgtcls_2.py b/config/bidirectional_tgtcls_2.py
index f1bf7bd..065b36c 100644
--- a/config/bidirectional_tgtcls_2.py
+++ b/config/bidirectional_tgtcls_2.py
@@ -26,8 +26,8 @@ embed_weights_init = IsotropicGaussian(0.01)
weights_init = IsotropicGaussian(0.1)
biases_init = Constant(0.01)
-batch_size = 300
-batch_sort_size = 20
+batch_size = 50
+batch_sort_size = 40
max_splits = 200