forked from huawei/mindspore2022
!10661 remove default word in kwargs
From: @sl_wang Reviewed-by: @kisnwang,@wang_zi_dong Signed-off-by: @wang_zi_dong
This commit is contained in:
commit
2dde52bfeb
|
|
@ -44,6 +44,6 @@ def create_network(name, *args, **kwargs):
|
|||
Create bert network for bert_thor.
|
||||
'''
|
||||
if name == 'bert_thor':
|
||||
is_training = kwargs.get("is_training", default=False)
|
||||
is_training = kwargs.get("is_training", False)
|
||||
return BertModel(bert_net_cfg, is_training, *args)
|
||||
raise NotImplementedError(f"{name} is not implemented in the repo")
|
||||
|
|
|
|||
Loading…
Reference in New Issue