Log some information about the input layer after initialization (#844)

Author: guillaumekln

opennmt/inputters/text_inputter.py

@@ -323,6 +323,10 @@ def _assert_is_initialized(self):
             )
 
 
+def _format_bool(b):
+    return "yes" if b else "no"
+
+
 class WordEmbedder(TextInputter):
     """Simple word embedder."""
 
@@ -394,6 +398,16 @@ def initialize(self, data_config):
             self.mark_start = sequence_controls.get("start")
             self.mark_end = sequence_controls.get("end")
 
+        tf.get_logger().info(
+            "Initialized %s input layer:", self.asset_prefix.strip("_")
+        )
+        tf.get_logger().info(" - vocabulary size: %d", self.vocabulary_size)
+        tf.get_logger().info(
+            " - special tokens: BOS=%s, EOS=%s",
+            _format_bool(self.mark_start),
+            _format_bool(self.mark_end),
+        )
+
     def make_features(self, element=None, features=None, training=None):
         """Converts words tokens to ids."""
         features = super().make_features(