-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Closed
Labels
questionFurther information is requestedFurther information is requested
Description
How to save the checkpoint only for the last epoch?
In the docs:
if save_top_k == k, the best k models according to the quantity monitored will be saved. if save_top_k == 0, no models are saved. if save_top_k == -1, all models are saved. Please note that the monitors are checked every period epochs. if save_top_k >= 2 and the callback is called multiple times inside an epoch, the name of the saved file will be appended with a version count starting with v0.
k = 0does not save any.k > 1saves only a few the bestk = -1saves all of them
Currently, I am using k = -1, but it space consuming.
Metadata
Metadata
Assignees
Labels
questionFurther information is requestedFurther information is requested