train_sweep_wiki_cs.yaml

program: train.py
project: "GRACE_recoverability_hps"
name: "wiki_cs_recoverability"
method: bayes
metric:
  goal: maximize
  name: F1Mi_mean
parameters:
  use_wandb:
    values: [true]
  dataset:
    values: ["wiki_cs"]
  method:
    values: ["recoverability"]
  wiki_cs.learning_rate:
    values: [1e-4, 5e-5, 1e-5, 5e-6, 1e-6]
  wiki_cs.num_hidden:
    values: [128, 256, 512]
  wiki_cs.num_layers:
    values: [2, 4]
  wiki_cs.kernel_lambda:
    values: [100, 10, 1, 0, 1e-2, 1e-3, 1e-4, 1e-5]
  wiki_cs.num_epochs:
    values: [ 100 ]

command:
  - ${env}
  - python
  - ${program}
  - ${args_no_hyphens}