[6ac965]: / conf / models / Torch_CRFNet_0_001.yaml

Download this file

13 lines (12 with data), 290 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
# Config for the Torch_XLearner model
# ==============================================================================
# Model parameters
n_layers_r: 1
n_layers_out: 1
n_units_out: 100
n_units_r: 100
batch_size: 256
n_iter: 2000
batch_norm: false
nonlin: "relu"
penalty_disc: 0.001 #0.001