checkpoint for best test accuracy actually works