Merge pull request #1 from Asjidkalam/master

Fixed Code Execution on tensorflow/models
tensorflow · Jan 25, 2021 · 02b4f87 · 02b4f87
2 parents efa0c44 + f6e9eda
commit 02b4f87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/official/modeling/hyperparams/params_dict.py b/official/modeling/hyperparams/params_dict.py
@@ -312,7 +312,7 @@ def _get_kvs(tokens, params_dict):
 def read_yaml_to_params_dict(file_path):
   """Reads a YAML file to a ParamsDict."""
   with tf.io.gfile.GFile(file_path, 'r') as f:
-    params_dict = yaml.load(f, Loader=yaml.FullLoader)
+    params_dict = yaml.load(f, Loader=yaml.SafeLoader)
     return ParamsDict(params_dict)