File size: 793 Bytes
ae1d5a9 |
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 30 31 32 33 |
lightning:
find_unused_parameters: false
modelcheckpoint:
params:
every_n_train_steps: 5000
save_top_k: -1
monitor: null
callbacks:
image_logger:
target: main.ImageLogger
params:
batch_frequency: 2000
max_images: 4
increase_log_steps: false
log_first_step: true
log_all_val: true
log_images_kwargs:
use_ema_scope: true
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
check_val_every_n_epoch: 10
gpus: 1
|