Skip to content
Discussion options

You must be logged in to vote

you can create a custom ModelCheckpoint instance with save_top_k=-1 and pass it in inside Trainer callbacks.

ckpt_callback = ModelCheckpoint(save_top_k=-1, ...)
trainer = Trainer(callbacks=[ckpt_callback], ...)

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@mcomunita
Comment options

Answer selected by mcomunita
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment