diff --git a/algorithmic_efficiency/spec.py b/algorithmic_efficiency/spec.py index ba48adbf6..648443530 100644 --- a/algorithmic_efficiency/spec.py +++ b/algorithmic_efficiency/spec.py @@ -374,8 +374,8 @@ def eval_model(self, data_dir=imagenet_v2_data_dir if imagenet_v2_data_dir else data_dir, global_step=global_step) for k, v in test_metrics.items(): - eval_metrics['test/' + k] = v - eval_metrics['test/num_examples'] = self.num_test_examples + eval_metrics['valid_test/' + k] = v + eval_metrics['validi_test/num_examples'] = self.num_test_examples + self.num_validation_examples except NotImplementedError: pass