mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix logging torch.dtype.

Browse Source
This commit is contained in:
Fangjun Kuang 2025-05-21 12:01:38 +08:00
parent e79833aad2
commit 2ca458bb56
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/utils.py
View File

@ -186,7 +186,7 @@ class AttributeDict(dict):
tmp = {} tmp = {}
for k, v in self.items(): for k, v in self.items():
# PosixPath is ont JSON serializable # PosixPath is ont JSON serializable
if isinstance(v, pathlib.Path) or isinstance(v, torch.device): if isinstance(v, (pathlib.Path, torch.device, torch.dtype)):
v = str(v) v = str(v)
tmp[k] = v tmp[k] = v
return json.dumps(tmp, indent=indent, sort_keys=True) return json.dumps(tmp, indent=indent, sort_keys=True)