From cd460f7bf174011ff65a5aec1b68f01a3496781e Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 7 May 2022 17:18:34 +0800 Subject: [PATCH] Stringify torch.__version__ before serializing it. (#354) --- icefall/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icefall/env.py b/icefall/env.py index c29cbb078..0192d1c11 100644 --- a/icefall/env.py +++ b/icefall/env.py @@ -95,7 +95,7 @@ def get_env_info() -> Dict[str, Any]: "k2-git-sha1": k2.version.__git_sha1__, "k2-git-date": k2.version.__git_date__, "lhotse-version": lhotse.__version__, - "torch-version": torch.__version__, + "torch-version": str(torch.__version__), "torch-cuda-available": torch.cuda.is_available(), "torch-cuda-version": torch.version.cuda, "python-version": sys.version[:3],