From 7c0070e6f6aa5c133805c7c7f2818691cb69d34b Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 8 Apr 2022 11:39:54 +0800
Subject: [PATCH] Display torch version in the training log. (#299)

---
 icefall/env.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/icefall/env.py b/icefall/env.py
index 0684c4bf1..c29cbb078 100644
--- a/icefall/env.py
+++ b/icefall/env.py
@@ -95,6 +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-cuda-available": torch.cuda.is_available(),
         "torch-cuda-version": torch.version.cuda,
         "python-version": sys.version[:3],