lightning:
  find_unused_parameters: false
  modelcheckpoint:
    params:
      monitor: train/loss_simple
      save_top_k: -1
      every_n_train_steps: 405
  callbacks:
    image_logger:
      target: main.ImageLogger
      params:
        batch_frequency: 10000000
        log_on_batch_idx: true
        max_images: 4
        increase_log_steps: false
        log_images_kwargs:
          inpaint: false
          plot_denoise_rows: false
          plot_progressive_rows: false
          plot_diffusion_rows: false
          'N': 4
          ddim_steps: 50
  trainer:
    benchmark: true
    max_steps: 20000
    check_val_every_n_epoch: 1000
    num_sanity_val_steps: 0
    accumulate_grad_batches: 8
    accelerator: gpu
    strategy: ddp
    num_nodes: 2
    gpus: 0,1,2,3,4,5,6,7