File size: 550 Bytes
c0ec2ff 3cc58a2 c0ec2ff 3cc58a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from dashboard_utils.time_tracker import simple_time_tracker, _log
import wandb
WANDB_REPO = "learning-at-home/Main_metrics"
@simple_time_tracker(_log)
def get_main_metrics():
api = wandb.Api()
runs = api.runs(WANDB_REPO)
run = runs[0]
history = run.scan_history(keys=["step", "loss", "alive peers"])
steps = []
losses = []
alive_peers = []
for row in history:
steps.append(row["step"])
losses.append(row["loss"])
alive_peers.append(row["alive peers"])
return steps, losses, alive_peers |