21 lines
314 B
YAML
21 lines
314 B
YAML
name: EQTransformer
|
|
method: bayes
|
|
metric:
|
|
goal: minimize
|
|
name: val_loss
|
|
parameters:
|
|
model_name:
|
|
value:
|
|
- EQTransformer
|
|
batch_size:
|
|
distribution: int_uniform
|
|
max: 1024
|
|
min: 256
|
|
max_epochs:
|
|
value:
|
|
- 30
|
|
learning_rate:
|
|
distribution: uniform
|
|
max: 0.02
|
|
min: 0.005
|