File size: 705 Bytes
40e1762
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
32
33
# @package _global_

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

defaults:
  - override /datamodule: focus150.yaml
  - override /model: focusLiteNN.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: "focusLiteNN_150"
seed: 12345

trainer:
  min_epochs: 1
  max_epochs: 100

model:
  pre_trained: False
  num_channel: 3
  lr: 0.001
  weight_decay: 0.0005

datamodule:
  batch_size: 128
  augmentation: True