Best trained model not correctly saving with pytorch

Question

i want to save the best model when training but it keeps saving the first trained model

#Train the model for 4 epochs
from collections import defaultdict
#Def