replace `step` by `optimizer_step`
Browse files
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=["
|
18 |
|
19 |
steps = []
|
20 |
losses = []
|
21 |
alive_peers = []
|
22 |
dates = []
|
23 |
for row in history:
|
24 |
-
steps.append(row["
|
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"]))
|