diff --git a/model_zoo/official/nlp/bert_thor/mindspore_hub_conf.py b/model_zoo/official/nlp/bert_thor/mindspore_hub_conf.py index 743cc4cbe92..05672d3f2e0 100644 --- a/model_zoo/official/nlp/bert_thor/mindspore_hub_conf.py +++ b/model_zoo/official/nlp/bert_thor/mindspore_hub_conf.py @@ -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")