vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-02-22 21:12:15 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Fix wandb logging (#42)

Browse source 
Co-authored-by: Srini Iyer <sviyer@meta.com>
This commit is contained in:
Srinivasan Iyer 2025-02-05 16:24:39 -08:00 committed by GitHub
parent c79b1fdbd0
commit 7cf8fab49b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
bytelatent/metrics.py
View file

@ -4,7 +4,6 @@
import json import json
import logging import logging
from collections import namedtuple from collections import namedtuple
from dataclasses import asdict
from datetime import datetime, timezone from datetime import datetime, timezone
from pathlib import Path from pathlib import Path
from typing import Any, Union from typing import Any, Union
@ -68,8 +67,8 @@ class MetricLogger:
and get_is_master() and get_is_master()
): ):
run = wandb.init( run = wandb.init(
config=asdict(self.args), config=self.args.model_dump(),
**asdict(self.args.logging.wandb), **self.args.logging.wandb.model_dump(),
) )
def log(self, metrics: dict[str, Any]): def log(self, metrics: dict[str, Any]):