seed: 3407

dataset:
  name: 'ReconData'
  args:
    data_root: '/data1/lxl/data/ocr/generate1108'
    train_anno_file: '/data1/lxl/data/ocr/generate1108/train.csv'
    val_anno_file: '/data1/lxl/data/ocr/generate1108/val.csv'
    fixed_size: 512

dataloader:
  batch_size: 8
  num_workers: 16
  pin_memory: true
  collate_fn: 'base_collate_fn'

model:
  name: 'Unet'
  args:
    encoder_name: 'resnet50'
    encoder_weights: 'imagenet'
    in_channels: 3
    classes: 3
    activation: 'tanh'

solver:
  name: 'BaseSolver'
  args:
    epoch: 30

  optimizer:
    name: 'Adam'
    args:
      lr: !!float 1e-4
      weight_decay: !!float 5e-5

  lr_scheduler:
    name: 'StepLR'
    args:
      step_size: 15
      gamma: 0.1

  loss:
    name: 'L1Loss'
    args:
      reduction: 'mean'

  logger:
    log_root: '/data1/lxl/code/ocr/removal/log'
    suffix: 'residual'

  metric:
    name: 'Recon'