How to Only Save the Best Model During Training

Hi！
I'm currently fine-tuning my model using train.py, but it seems like the best model isn't being saved and is continually being overwritten. Is there any code to ensure that only the best model is saved in the end?