diff --git a/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/bin/train.py b/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/bin/train.py
index a7ba284003460dd08e3d68511f7313e1b0687830..2995636e2c4138a0e0dc58d6c6b65215080a7a72 100644
--- a/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/bin/train.py
+++ b/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/bin/train.py
@@ -85,7 +85,7 @@ def load_config_and_update_flags():
       config_path = os.path.abspath(config_path)
       tf.logging.info("Loading config from %s", config_path)
       with tf.gfile.GFile(config_path.strip()) as config_file:
-        config_flags = yaml.load(config_file)
+        config_flags = yaml.safe_load(config_file)
         final_config = _deep_merge_dict(final_config, config_flags)
 
   tf.logging.info("Final Config:\n%s", yaml.dump(final_config))
diff --git a/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/configurable.py b/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/configurable.py
index efc5e9631be2bba28ef6f7f4eeca6b34abff36c2..4263eb8b980e89b2506560bf23c8a5566d13c499 100644
--- a/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/configurable.py
+++ b/TensorFlow/built-in/nlp/Transformer_ID0004_for_TensorFlow/noahnmt/configurable.py
@@ -80,7 +80,7 @@ def _maybe_load_yaml(item):
   it is returned as-is.
   """
   if isinstance(item, six.string_types):
-    return yaml.load(item)
+    return yaml.safe_load(item)
   elif isinstance(item, dict):
     return item
   else: