program: Source/hyperparameter_tuning.py
method: grid
metric:
name: val_loss
goal: minimize
parameters:
num_heads:
values: [2, 3, 4]
EGAT_Layers:
values: [2, 3, 4]
hidden_dim:
values: [32, 64, 128]
command:
- ${env}
- ${interpreter}
- ${program}
- "--config"
- "Example/Molecule/QM9.yaml"
- "--data_path"
- "Example/Generated_Data"
- "--save_path"
- "${args}"
I was thinking about something like this
- "--data_path"
- "Example/Generated_Data/num_heads_${num_heads}_EGAT_${EGAT_Layers}_hidden_${hidden_dim}"
- "--save_path"
- "Example/Generated_Data/num_heads_${num_heads}_EGAT_${EGAT_Layers}_hidden_${hidden_dim}"
Does this make sense?