Log test set stats while learning