| 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 | |