dreamedit_official_ckpt/can-ldj/configs/can2023-04-17T01-38-40-lightning.yaml

lightning:
  modelcheckpoint:
    params:
      every_n_train_steps: 500
  callbacks:
    image_logger:
      target: main.ImageLogger
      params:
        batch_frequency: 500
        max_images: 8
        increase_log_steps: false
  trainer:
    benchmark: true
    max_steps: 800
    accelerator: ddp
    gpus: 5,