| lightning: | |
| find_unused_parameters: false | |
| modelcheckpoint: | |
| params: | |
| every_n_train_steps: 5000 | |
| callbacks: | |
| image_logger: | |
| target: main.ImageLogger | |
| params: | |
| batch_frequency: 500 | |
| max_images: 4 | |
| increase_log_steps: false | |
| log_first_step: false | |
| log_images_kwargs: | |
| use_ema_scope: false | |
| inpaint: false | |
| plot_progressive_rows: false | |
| plot_diffusion_rows: false | |
| 'N': 4 | |
| unconditional_guidance_scale: 3.0 | |
| unconditional_guidance_label: | |
| - '' | |
| trainer: | |
| benchmark: true | |
| num_sanity_val_steps: 0 | |
| accumulate_grad_batches: 1 | |
| accelerator: ddp | |
| gpus: 4,5 | |