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
|