File size: 697 Bytes
389784b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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
|