Dashboard / dashboard_utils /main_metrics.py
SaulLu's picture
Add the possibility to choose the run taken for the main metric
11dce67
raw
history blame
762 Bytes
import datetime
import streamlit as st
import wandb
from dashboard_utils.time_tracker import _log, simple_time_tracker
WANDB_RUN_URL = st.secrets["WANDB_RUN_URL_MAIN_METRICS"]
CACHE_TTL = 100
@st.cache(ttl=CACHE_TTL, show_spinner=False)
@simple_time_tracker(_log)
def get_main_metrics():
api = wandb.Api()
run = api.run(WANDB_RUN_URL)
history = run.scan_history(keys=["step", "loss", "alive peers", "_timestamp"])
steps = []
losses = []
alive_peers = []
dates = []
for row in history:
steps.append(row["step"])
losses.append(row["loss"])
alive_peers.append(row["alive peers"])
dates.append(datetime.datetime.utcfromtimestamp(row["_timestamp"]))
return steps, dates, losses, alive_peers