configs/experiment/focusConvMSE_150.yaml

# @package _global_

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

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

trainer:
  min_epochs: 1
  max_epochs: 100

model:
  image_size: 150
  pool_size: 2
  conv1_size: 5
  conv1_channels: 6
  conv2_size: 5
  conv2_channels: 16
  lin1_size: 100
  lin2_size: 80
  output_size: 1
  lr: 0.001
  weight_decay: 0.0005

datamodule:
  batch_size: 128