How to Only Save the Best Model During Training

ai-forever / ghost

A new one shot face swap approach for image and video domains

Apache License 2.0

1.25k stars 264 forks source link

How to Only Save the Best Model During Training #91

Open ShotaYamasaki opened 9 months ago

ShotaYamasaki commented 9 months ago

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?