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:
19
experiments/sweep_basicphase_ae.yaml
Normal file
19
experiments/sweep_basicphase_ae.yaml
Normal 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
|
20
experiments/sweep_eqtransformer.yaml
Normal file
20
experiments/sweep_eqtransformer.yaml
Normal 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
|
@@ -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
|
||||
|
@@ -13,7 +13,7 @@ parameters:
|
||||
min: 256
|
||||
max_epochs:
|
||||
value:
|
||||
- 15
|
||||
- 30
|
||||
learning_rate:
|
||||
distribution: uniform
|
||||
max: 0.02
|
||||
|
Reference in New Issue
Block a user