39 lines
638 B
YAML
39 lines
638 B
YAML
|
method: bayes
|
||
|
|
metric:
|
||
|
|
goal: minimize
|
||
|
|
name: test_loss
|
||
|
|
parameters:
|
||
|
|
batch_size:
|
||
|
|
distribution: int_uniform
|
||
|
|
max: 512
|
||
|
|
min: 128
|
||
|
|
dataset:
|
||
|
|
distribution: categorical
|
||
|
|
values:
|
||
|
|
- igf_1
|
||
|
|
epochs:
|
||
|
|
distribution: int_uniform
|
||
|
|
max: 60
|
||
|
|
min: 15
|
||
|
|
learning_rate:
|
||
|
|
distribution: uniform
|
||
|
|
max: 0.02
|
||
|
|
min: 0.005
|
||
|
|
model_name:
|
||
|
|
distribution: categorical
|
||
|
|
values:
|
||
|
|
- PhaseNet
|
||
|
|
pretrained:
|
||
|
|
distribution: categorical
|
||
|
|
values:
|
||
|
|
- diting
|
||
|
|
- ethz
|
||
|
|
- geofon
|
||
|
|
- instance
|
||
|
|
- iquique
|
||
|
|
- lendb
|
||
|
|
- neic
|
||
|
|
- original
|
||
|
|
- scedc
|
||
|
|
program: training_wandb_sweep.py
|