"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "1e8c0fc3a73a73afa9d3ff2fc46ed1fddc9ff6cc"
Commit 8454dedc authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 371418693
parent ba14f96a
...@@ -458,7 +458,7 @@ def override_params_dict(params, dict_or_string_or_yaml_file, is_strict): ...@@ -458,7 +458,7 @@ def override_params_dict(params, dict_or_string_or_yaml_file, is_strict):
params.override(params_dict, is_strict) params.override(params_dict, is_strict)
else: else:
with tf.io.gfile.GFile(dict_or_string_or_yaml_file) as f: with tf.io.gfile.GFile(dict_or_string_or_yaml_file) as f:
params.override(yaml.load(f, Loader=LOADER), is_strict) params.override(yaml.load(f, Loader=yaml.FullLoader), is_strict)
else: else:
raise ValueError('Unknown input type to parse.') raise ValueError('Unknown input type to parse.')
return params return params
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