train_cldm.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/cldm.yaml
  # Path to the checkpoints or weights you want to resume. At the begining, 
  # this should be set to the initial weights created by scripts/make_stage2_init_weight.py.
  resume: weights/init.pth

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

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