How to set the model to just save top_k checkpoints?

LambdaLabsML / examples

Deep Learning Examples

MIT License

805 stars 103 forks source link

Closed zhaoshitian closed 2 years ago

zhaoshitian commented 2 years ago

It seems that every 2000 steps the checkpoints will be saved, but I just want to save top_3(metric=val_loss). How can I implement it?

zhaoshitian commented 2 years ago

I changed some code in the file 'pokemon.yaml' like the picture above, and succeed.