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"]))
|