Update MonitorBasedSaveBest to store best model in memory