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]