Gil-Simas commited on
Commit
363365d
·
1 Parent(s): 4e85605

remove model from log_key

Browse files
Files changed (1) hide show
  1. user-friendly-metrics.py +14 -15
user-friendly-metrics.py CHANGED
@@ -176,21 +176,20 @@ class UserFriendlyMetrics(evaluate.Metric):
176
 
177
  # Log overall metrics
178
  if "overall" in result:
179
- for overall_key, overall_metrics in result["overall"]["all"].items():
180
- for metric, value in overall_metrics.items():
181
- log_key = (
182
- f"{wandb_section}/overall/{overall_key}/{metric}"
183
- if wandb_section
184
- else f"overall/{overall_key}/{metric}"
185
- )
186
- run.log({log_key: value})
187
-
188
- if debug:
189
- print(f" {log_key} = {value}")
190
-
191
- for category, metrics in categories.items():
192
- if metric in metrics:
193
- chart_data[category].append([metric, value])
194
  print("----------------------------------------------------")
195
 
196
  if log_plots:
 
176
 
177
  # Log overall metrics
178
  if "overall" in result:
179
+ for metric, value in overall_metrics["overall"]["all"].items():
180
+ log_key = (
181
+ f"{wandb_section}/overall/{metric}"
182
+ if wandb_section
183
+ else f"overall/{metric}"
184
+ )
185
+ run.log({log_key: value})
186
+
187
+ if debug:
188
+ print(f" {log_key} = {value}")
189
+
190
+ for category, metrics in categories.items():
191
+ if metric in metrics:
192
+ chart_data[category].append([metric, value])
 
193
  print("----------------------------------------------------")
194
 
195
  if log_plots: