File size: 311 Bytes
80d4918 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
seed: 42
train:
batch_size: 32
epochs: 20
hidden_size: 64
dropout_prob: 0.13
learning_rate: 0.0075
l2_reg_lambda: 0.001
optimizer: Adam
loss: CrossEntropyLoss
tune:
n_trials: 50
hidden_size_choices: [32,64,128,256,512]
dropout_prob_range: [0.1, 0.5]
learning_rate_range: [1e-5, 1e-2]
|