brick_model / previous_logs /2022-10-19T11-50-36_lego /configs /2022-10-19T11-50-36-lightning.yaml
bolee's picture
add previous_logs
bd41086
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