Hannes Kuchelmeister commited on
Commit
b04e419
1 Parent(s): 12319ae

add experiment to compare MSE and MAE (focus150.yml and focusMSE_150.yml)

Browse files
configs/experiment/focus150.yaml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # @package _global_
2
+
3
+ # to execute this experiment run:
4
+ # python train.py experiment=example
5
+
6
+ defaults:
7
+ - override /datamodule: focus150.yaml
8
+ - override /model: focus150.yaml
9
+ - override /callbacks: default.yaml
10
+ - override /logger: tensorboard
11
+ - override /trainer: default.yaml
12
+
13
+ # all parameters below will be merged with parameters from default configurations set above
14
+ # this allows you to overwrite only specified parameters
15
+
16
+ # name of the run determines folder name in logs
17
+ name: "focusMAE"
18
+ seed: 12345
19
+
20
+ trainer:
21
+ min_epochs: 1
22
+ max_epochs: 100
23
+
24
+ model:
25
+ lin1_size: 128
26
+ lin2_size: 256
27
+ lin3_size: 64
28
+
29
+ datamodule:
30
+ batch_size: 128
31
+
configs/experiment/focusMSE_150.yaml ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # @package _global_
2
+
3
+ # to execute this experiment run:
4
+ # python train.py experiment=example
5
+
6
+ defaults:
7
+ - override /datamodule: focus150.yaml
8
+ - override /model: focus150.yaml
9
+ - override /callbacks: default.yaml
10
+ - override /logger: tensorboard
11
+ - override /trainer: default.yaml
12
+
13
+ # all parameters below will be merged with parameters from default configurations set above
14
+ # this allows you to overwrite only specified parameters
15
+
16
+ # name of the run determines folder name in logs
17
+ name: "focusMSE_150"
18
+ seed: 12345
19
+
20
+ trainer:
21
+ min_epochs: 1
22
+ max_epochs: 100
23
+
24
+ model:
25
+ _target_: src.models.focus_module.FocusMSELitModule
26
+ lin1_size: 128
27
+ lin2_size: 256
28
+ lin3_size: 64
29
+
30
+ datamodule:
31
+ batch_size: 128
32
+