From 7cf8fab49bc75737c73d050e1f3bb85fd826e00c Mon Sep 17 00:00:00 2001 From: Srinivasan Iyer Date: Wed, 5 Feb 2025 16:24:39 -0800 Subject: [PATCH] Fix wandb logging (#42) Co-authored-by: Srini Iyer --- bytelatent/metrics.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/bytelatent/metrics.py b/bytelatent/metrics.py index 77dc4d7..e746e4f 100644 --- a/bytelatent/metrics.py +++ b/bytelatent/metrics.py @@ -4,7 +4,6 @@ import json import logging from collections import namedtuple -from dataclasses import asdict from datetime import datetime, timezone from pathlib import Path from typing import Any, Union @@ -68,8 +67,8 @@ class MetricLogger: and get_is_master() ): run = wandb.init( - config=asdict(self.args), - **asdict(self.args.logging.wandb), + config=self.args.model_dump(), + **self.args.logging.wandb.model_dump(), ) def log(self, metrics: dict[str, Any]):