Added logging MAE for the best runs, option to run a pipeline on a specific dataset, template bash scripts, GPLv3 license. Modified behavior of generating eval targets, it is skipped if targets already exist.

This commit is contained in:
Krystyna Milian
2023-10-12 14:27:53 +02:00
parent aa39980573
commit 4c2679a005
18 changed files with 310 additions and 368 deletions

View File

@@ -0,0 +1,19 @@
method: bayes
metric:
goal: minimize
name: val_loss
parameters:
model_name:
value:
- BasicPhaseAE
batch_size:
distribution: int_uniform
max: 1024
min: 256
max_epochs:
value:
- 20
learning_rate:
distribution: uniform
max: 0.02
min: 0.001

View File

@@ -0,0 +1,20 @@
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

View File

@@ -13,14 +13,14 @@ parameters:
min: 256
max_epochs:
value:
- 3
- 30
learning_rate:
distribution: uniform
max: 0.02
min: 0.005
highpass:
value:
- 2
- 1
lowpass:
value:
- 10
- 10

View File

@@ -13,7 +13,7 @@ parameters:
min: 256
max_epochs:
value:
- 15
- 30
learning_rate:
distribution: uniform
max: 0.02