xla

c4f507c1 - Test max accuracy instead of final accuracy (#1438)

Test max accuracy instead of final accuracy (#1438) * assert against max accuracy instead of final accuracy * save max_accuracy instead of all accuracies * remove print for new best acc