ckpt every 5k
This commit is contained in:
parent
3809ba7046
commit
f5bf534f42
1 changed files with 4 additions and 0 deletions
|
@ -105,6 +105,10 @@ data:
|
||||||
lightning:
|
lightning:
|
||||||
find_unused_parameters: False
|
find_unused_parameters: False
|
||||||
|
|
||||||
|
modelcheckpoint:
|
||||||
|
params:
|
||||||
|
every_n_train_steps: 5000
|
||||||
|
|
||||||
callbacks:
|
callbacks:
|
||||||
image_logger:
|
image_logger:
|
||||||
target: main.ImageLogger
|
target: main.ImageLogger
|
||||||
|
|
Loading…
Reference in a new issue