diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-25 14:19:40 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-25 14:19:40 -0400 |
| commit | a557b939eb104ec7e0df42193e014e3137eb70f8 (patch) | |
| tree | bbff025fb98dcd8d992bbdbc590fbd32f7227803 /config/bidirectional_tgtcls_2.py | |
| parent | 5e216e8c5197568c39029793ebafc4df50215cae (diff) | |
| download | taxi-a557b939eb104ec7e0df42193e014e3137eb70f8.tar.gz taxi-a557b939eb104ec7e0df42193e014e3137eb70f8.zip | |
Change batch size for smaller GPU
Diffstat (limited to 'config/bidirectional_tgtcls_2.py')
| -rw-r--r-- | config/bidirectional_tgtcls_2.py | 4 |
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 |