Skip to content

GitLab

Unverified Commit 02b4f871 authored by Jamie Slome's avatar Jamie Slome Committed by GitHub
Browse files

Merge pull request #1 from Asjidkalam/master 

Fixed Code Execution on tensorflow/models
parents efa0c440 f6e9eda8
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/hyperparams/params_dict.py
View file @ 02b4f871
...@@ -312,7 +312,7 @@ class ParamsDict(object): ...@@ -312,7 +312,7 @@ class ParamsDict(object):
def read_yaml_to_params_dict(file_path): def read_yaml_to_params_dict(file_path):
"""Reads a YAML file to a ParamsDict.""" """Reads a YAML file to a ParamsDict."""
with tf.io.gfile.GFile(file_path, 'r') as f: 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) return ParamsDict(params_dict)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment