Spaces:

lucalp
/

blt-entropy-patcher

Running on Zero

Srinivasan Iyer

sviyer commited on Feb 6

Commit

7cf8fab

unverified ·

1 Parent(s): c79b1fd

Fix wandb logging (#42)

Co-authored-by: Srini Iyer <sviyer@meta.com>

Files changed (1) hide show

bytelatent/metrics.py CHANGED Viewed

@@ -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),
             )
     def log(self, metrics: dict[str, Any]):

 import json
 import logging
 from collections import namedtuple
 from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Union
             and get_is_master()
         ):
             run = wandb.init(
+                config=self.args.model_dump(),
+                **self.args.logging.wandb.model_dump(),
             )
     def log(self, metrics: dict[str, Any]):