File size: 243 Bytes
3af7a55 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
lightning:
callbacks:
image_logger:
target: train.ImageLogger
params:
batch_frequency: 1000
max_images: 8
increase_log_steps: false
trainer:
max_steps: 550
accelerator: ddp
gpus: 0,1,2,3
|