speedup
This commit is contained in:
parent
de1470f43b
commit
ae05402bc6
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue