diff --git a/python/examples/bert/prepare_model.py b/python/examples/bert/prepare_model.py index 521aea9ae2e892c4e071ef2e9917e51f8e626743..e883b6b15746946a7f3412fe64c5933c4cfb37ab 100644 --- a/python/examples/bert/prepare_model.py +++ b/python/examples/bert/prepare_model.py @@ -20,7 +20,7 @@ import paddle paddle.enable_static() model_name = "bert_chinese_L-12_H-768_A-12" -module = hub.Module(model_name) +module = hub.Module(name=model_name) inputs, outputs, program = module.context( trainable=True, max_seq_len=int(sys.argv[1])) place = fluid.core_avx.CPUPlace()