train_swinir.yaml

data:
  target: dataset.data_module.BIRDataModule
  params:
    # Path to training set configuration file.
    train_config: configs/dataset/general_deg_realesrgan_train.yaml
    # Path to validation set configuration file.
    val_config: configs/dataset/general_deg_realesrgan_val.yaml

model:
  # You can set learning rate in the following configuration file.
  config: configs/model/swinir.yaml
  # Path to the checkpoints or weights you want to resume.
  resume: ~

lightning:
  seed: 231
  
  trainer:
    accelerator: ddp
    precision: 32
    # Indices of GPUs used for training.
    gpus: [5,7]
    # Path to save logs and checkpoints.
    default_root_dir: stage1_checkpoints
    # Max number of training steps (batches).
    max_steps: 10000
    # Validation frequency in terms of training steps.
    val_check_interval: 0.25
    # Log frequency of tensorboard logger.
    log_every_n_steps: 100
    # Accumulate gradients from multiple batches so as to increase batch size.
    accumulate_grad_batches: 2
  
  callbacks:
    - target: model.callbacks.ImageLogger
      params:
        # Log frequency of image logger.
        log_every_n_steps: 1000
        max_images_each_step: 2
        log_images_kwargs: ~

    - target: model.callbacks.ModelCheckpoint
      params:
        # Frequency of saving checkpoints.
        every_n_train_steps: 1000
        save_top_k: -1
        filename: "{step}"