Update task_iflytek_bert_of_theseus.py

bojone · Jul 17, 2020 · 9f6dd86 · 9f6dd86
1 parent a1d66e9
commit 9f6dd86
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/examples/task_iflytek_bert_of_theseus.py b/examples/task_iflytek_bert_of_theseus.py
@@ -149,20 +149,19 @@ def on_epoch_end(self, epoch, logs=None):
 predecessor = build_transformer_model(
     config_path=config_path,
     checkpoint_path=checkpoint_path,
-    return_keras_model=False
+    return_keras_model=False,
+    prefix='Predecessor-'
 )
 
 # 加载预训练模型（3层）
 successor = build_transformer_model(
     config_path=config_path,
     checkpoint_path=checkpoint_path,
     return_keras_model=False,
-    num_hidden_layers=3
+    num_hidden_layers=3,
+    prefix='Successor-'
 )
 
-for layer in successor.model.layers:
-    layer.name = 'Successor-' + layer.name
-
 # 判别模型
 x_in = Input(shape=K.int_shape(predecessor.output)[1:])
 x = Lambda(lambda x: x[:, 0])(x_in)