File size: 655 Bytes
b04e419
 
 
 
 
 
 
 
 
eac0701
b04e419
 
 
 
 
 
41a3c26
b04e419
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# @package _global_

# to execute this experiment run:
# python train.py experiment=example

defaults:
  - override /datamodule: focus150.yaml
  - override /model: focus150.yaml
  - override /callbacks: default.yaml
  - override /logger: many_loggers
  - override /trainer: default.yaml

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

# name of the run determines folder name in logs
name: "focusMAE_150"
seed: 12345

trainer:
  min_epochs: 1
  max_epochs: 100

model:
  lin1_size: 128
  lin2_size: 256
  lin3_size: 64

datamodule:
  batch_size: 128