epos-ai/platform-demo-scripts
3
0
Fork 0
Code Issues Pull Requests Activity

initial commit

Browse Source
This commit is contained in:
Krystyna Milian
2023-07-05 09:58:06 +02:00
commit 128d657e76
15 changed files with 11868 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
experiments/config.json Normal file
View File

@@ -0,0 +1,11 @@
{
"epochs": 10,
"batch_size": 256,
"dataset": "igf_1",
"sampling_rate": 100,
"model_names": "EQTransformer,BasicPhaseAE,GPD",
"model_name": "PhaseNet",
"learning_rate": 0.01,
"pretrained": null,
"sampling_rate": 100
}

38
experiments/sweep.yaml Normal file
View File

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

32
experiments/sweep4.yaml Normal file
View File

@@ -0,0 +1,32 @@
method: bayes
metric:
goal: minimize
name: test_loss
parameters:
batch_size:
distribution: int_uniform
max: 512
min: 256
dataset:
distribution: categorical
values:
- igf_1
epochs:
distribution: categorical
values:
- 10
learning_rate:
distribution: uniform
max: 0.02
min: 0.01
model_name:
distribution: categorical
values:
# - EQTransformer
- PhaseNet
pretrained:
distribution: categorical
values:
- instance
- iquique
- false