From f6e9eda85f7e764e54199169ed829b94bcd07871 Mon Sep 17 00:00:00 2001 From: Asjid Kalam Date: Mon, 21 Dec 2020 16:21:41 +0530 Subject: [PATCH] fixed code execution --- official/modeling/hyperparams/params_dict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official/modeling/hyperparams/params_dict.py b/official/modeling/hyperparams/params_dict.py index b2061ca3c08..fa40a283692 100644 --- 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)