From f4f2d6f07e225a254659939d12b1f9cb745e260c Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Sat, 2 Oct 2021 10:22:18 +0000
Subject: [PATCH] print deps module version

---
 deepspeech/training/trainer.py |  2 ++
 deepspeech/utils/utility.py    | 21 ++++++++++++++++++++-
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py
index 35b1690b..75652ead 100644
--- a/deepspeech/training/trainer.py
+++ b/deepspeech/training/trainer.py
@@ -27,6 +27,7 @@ from deepspeech.utils import mp_tools
 from deepspeech.utils import profiler
 from deepspeech.utils.checkpoint import Checkpoint
 from deepspeech.utils.log import Log
+from deepspeech.utils.utility import all_version
 from deepspeech.utils.utility import seed_all
 from deepspeech.utils.utility import UpdateConfig
 
@@ -103,6 +104,7 @@ class Trainer():
         self.epoch = 0
         self.rank = dist.get_rank()
 
+        all_version()
         logger.info(f"Rank: {self.rank}/{dist.get_world_size()}")
 
         if args.seed:
diff --git a/deepspeech/utils/utility.py b/deepspeech/utils/utility.py
index 6f84c41b..159b686e 100644
--- a/deepspeech/utils/utility.py
+++ b/deepspeech/utils/utility.py
@@ -16,17 +16,36 @@ import distutils.util
 import math
 import os
 import random
+import sys
 from contextlib import contextmanager
 from typing import List
 
 import numpy as np
 import paddle
+import soundfile
+
+from deepspeech.utils.log import Log
+
+logger = Log(__name__).getlog()
 
 __all__ = [
-    "UpdateConfig", "seed_all", 'print_arguments', 'add_arguments', "log_add"
+    "all_version", "UpdateConfig", "seed_all", 'print_arguments',
+    'add_arguments', "log_add"
 ]
 
 
+def all_version():
+    vers = {
+        "python": sys.version,
+        "paddle": paddle.__version__,
+        "paddle_commit": paddle.version.commit,
+        "soundfile": soundfile.__version__,
+    }
+    logger.info("Deps Module Version:")
+    for k, v in vers.items():
+        logger.info(f"{k}: {v}")
+
+
 @contextmanager
 def UpdateConfig(config):
     """Update yacs config"""