albert_funnel_sweep.yaml

# controller:
#     type: local

entity: ${WANDB_ENTITY}
project: ${WANDB_PROJECT}

method: grid
metric:
  name: val_loss
  goal: minimize

command:
    - ${env}
    - python
    - albert_funnel.py
    - ${args}

parameters:
  data_number:
    values: [0]
  data_path:
    values: ["data/pheme5_lc"]
  LM_name:
    values: [
            "albert-large-v2",
            "funnel-transformer/xlarge-base"
            ]