fix

mlcommons · Nov 3, 2023 · f1756d9 · f1756d9
1 parent 760abdf
commit f1756d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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