Tensorflow save the best model