wandb_sweep_config.yml

command:
  - ${env}
  - ${interpreter}
  - train_model_on_features.py
  - ${args}
method: grid
metric:
  name: val_loss
  goal: minimize
parameters:
  data_base_path:
    value: ..\robot_or_brain_combined_data\
  epochs:
    value: 50
  batch_size:
    value: 16
  learning_rate:
    values:
      - 0.0003
      - 0.001
      - 0.003
      - 0.01
  lr_decay:
    value: 1e-4
  dropout_rate:
    value: 0.2
  use_augmentation:
    value: False