lightning:
  modelcheckpoint:
    params:
      every_n_train_steps: 1600
      save_top_k: -1
      save_on_train_epoch_end: false
  callbacks:
    metrics_over_trainsteps_checkpoint:
      params:
        every_n_train_steps: 25000
    image_logger:
      target: main.ImageLogger
      params:
        disabled: false
        enable_autocast: false
        batch_frequency: 5000
        max_images: 8
        increase_log_steps: false
        log_first_step: false
        log_images_kwargs:
          use_ema_scope: false
          'N': 1
          n_rows: 2
  trainer:
    devices: 0,1,2,3
    benchmark: true
    num_sanity_val_steps: 0
    accumulate_grad_batches: 1
    max_steps: 1610
    val_check_interval: 400
    accelerator: gpu