sweep_grid.yaml

program: ./3.Classification/ourModel/rnn_classifier_mergedData.py
method: grid
metric:
  name: Test accuracy
  goal: maximize
parameters:
  hidden_size:
    values:
    - 1000
    - 2000
  lrn_rate:
    distribution: constant
    value: 0.0000001
  batch_size:
    values:
    - 32
    - 64
  num_layers:
    values: 
    - 1
    - 2
  dropout:
    values:
    - 0.0
    - 0.5
command:
  - ${env}
  - python
  - ${program}
  - "--pose"
  - "--hands"
  - "--face"
  - "--wandb"
  - ${args}