Skip to content

GitLab

Commit 8b834f16 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Merge pull request #9669 from 418sec:1-other-models 

PiperOrigin-RevId: 353731112
parents 9d5eb798 02b4f871
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/modeling/hyperparams/params_dict.py
View file @ 8b834f16
...@@ -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