diff options
Diffstat (limited to 'config/dest_simple_mlp_tgtcls_1_cswdtx.py')
-rw-r--r-- | config/dest_simple_mlp_tgtcls_1_cswdtx.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/config/dest_simple_mlp_tgtcls_1_cswdtx.py b/config/dest_simple_mlp_tgtcls_1_cswdtx.py index de076ef..f350457 100644 --- a/config/dest_simple_mlp_tgtcls_1_cswdtx.py +++ b/config/dest_simple_mlp_tgtcls_1_cswdtx.py @@ -33,6 +33,10 @@ mlp_biases_init = Constant(0.001) learning_rate = 0.0001 momentum = 0.99 -batch_size = 32 +batch_size = 100 + +use_cuts_for_training = True +max_splits = 1 valid_set = 'cuts/test_times_0' + |