SaulLu commited on
Commit
78a019f
·
1 Parent(s): 102ea52

replace `step` by `optimizer_step`

Browse files
Files changed (1) hide show
  1. dashboard_utils/main_metrics.py +2 -2
dashboard_utils/main_metrics.py CHANGED
@@ -14,14 +14,14 @@ CACHE_TTL = 100
14
  def get_main_metrics():
15
  api = wandb.Api()
16
  run = api.run(WANDB_RUN_URL)
17
- history = run.scan_history(keys=["step", "loss", "alive peers", "_timestamp"])
18
 
19
  steps = []
20
  losses = []
21
  alive_peers = []
22
  dates = []
23
  for row in history:
24
- steps.append(row["step"])
25
  losses.append(row["loss"])
26
  alive_peers.append(row["alive peers"])
27
  dates.append(datetime.datetime.utcfromtimestamp(row["_timestamp"]))
 
14
  def get_main_metrics():
15
  api = wandb.Api()
16
  run = api.run(WANDB_RUN_URL)
17
+ history = run.scan_history(keys=["optimizer_step", "loss", "alive peers", "_timestamp"])
18
 
19
  steps = []
20
  losses = []
21
  alive_peers = []
22
  dates = []
23
  for row in history:
24
+ steps.append(row["optimizer_step"])
25
  losses.append(row["loss"])
26
  alive_peers.append(row["alive peers"])
27
  dates.append(datetime.datetime.utcfromtimestamp(row["_timestamp"]))