File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -890,11 +890,11 @@ def _save_metrics(self) -> None:
890890 """Save metrics results to self.ckpt_save_dir/test_metrics.json"""
891891
892892 metrics_results = {}
893- metrics_results ["overall" ] = {k : self .meter_pool .get_value (f"test/{ k } " ) for k in self .metrics . keys () }
893+ metrics_results ["overall" ] = {k : self .meter_pool .get_value (f"test/{ k } " ) for k in self .metrics }
894894 if self .eval_horizons is not None :
895895 for i in self .eval_horizons :
896896 metrics_results [f"horizon_{ i + 1 } " ] = {
897- k : self .meter_pool .get_value (f"test/{ k } @h{ i + 1 } " ) for k in self .metrics . keys () }
897+ k : self .meter_pool .get_value (f"test/{ k } @h{ i + 1 } " ) for k in self .metrics }
898898 with open (os .path .join (self .ckpt_save_dir , "test_metrics.json" ), "w" ) as f :
899899 json .dump (metrics_results , f , indent = 4 )
900900 metrics_path = os .path .join (self .ckpt_save_dir , "test_metrics.json" )
You can’t perform that action at this time.
0 commit comments