"...targets/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "3924f7ee749a91287053ce466923ad83705283b4"
Unverified Commit 5ee40948 authored by xiaoting's avatar xiaoting Committed by GitHub
Browse files

Merge pull request #545 from tink2123/fix_bug_for_srn

fix bug
parents c49be2da 9c59ae1c
......@@ -58,7 +58,7 @@ class RecModel(object):
self.loss_type = global_params['loss_type']
self.image_shape = global_params['image_shape']
self.max_text_length = global_params['max_text_length']
if "num_heads" in params:
if "num_heads" in global_params:
self.num_heads = global_params["num_heads"]
else:
self.num_heads = None
......
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