File size: 318 Bytes
1fb0501 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
lightning:
modelcheckpoint:
params:
every_n_train_steps: 500
callbacks:
image_logger:
target: main.ImageLogger
params:
batch_frequency: 500
max_images: 8
increase_log_steps: false
trainer:
benchmark: true
max_steps: 800
accelerator: ddp
gpus: 0,
|