This commit is contained in:
harubaru 2022-09-23 15:26:58 -07:00 committed by GitHub
parent de1470f43b
commit ae05402bc6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions

View File

@ -89,15 +89,16 @@ data:
val_split: 64 val_split: 64
lightning: lightning:
find_unused_parameters: False
modelcheckpoint: modelcheckpoint:
params: params:
every_n_train_steps: 500 every_n_train_steps: 2000
callbacks: callbacks:
image_logger: image_logger:
target: main.ImageLogger target: main.ImageLogger
params: params:
batch_frequency: 500 batch_frequency: 2000
max_images: 4 max_images: 2
increase_log_steps: False increase_log_steps: False
log_first_step: False log_first_step: False
log_images_kwargs: log_images_kwargs: