From ad7727b4e6f901843b5a0f9d740b51f912c7e0be Mon Sep 17 00:00:00 2001 From: wangjiawei04 Date: Mon, 8 Feb 2021 12:42:47 +0000 Subject: [PATCH] fix bert prepare_model.py --- python/examples/bert/prepare_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/examples/bert/prepare_model.py b/python/examples/bert/prepare_model.py index 521aea9a..e883b6b1 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() -- GitLab