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

Fix logging torch.dtype. (#1947)

Browse Source
This commit is contained in:
Fangjun Kuang 2025-05-21 12:04:57 +08:00 committed by GitHub
parent e79833aad2
commit fd8f8780fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 = {}
for k, v in self.items():
# 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)
tmp[k] = v
return json.dumps(tmp, indent=indent, sort_keys=True)