fix convert

liuzh47 · Aug 11, 2020 · 93bd659 · 93bd659
1 parent 9238d56
commit 93bd659
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/conversion_toolkits/convert_tf_hub_model.py b/scripts/conversion_toolkits/convert_tf_hub_model.py
@@ -169,9 +169,9 @@ def convert_tf_assets(tf_assets_dir, model_type):
     ('predictions/output_bias', 'mlm_decoder.3.bias'),
     ('transformer/layer_', 'encoder.all_layers.'),
     ('word_embeddings/embeddings', 'word_embed.weight'),
-    ('embedding_postprocessor/type_embeddings', 'token_type_embed.weight'),
-    ('embedding_postprocessor/position_embeddings', 'token_pos_embed._embed.weight'),
-    ('embedding_postprocessor/layer_norm', 'embed_layer_norm'),
+    ('type_embeddings/embeddings', 'token_type_embed.weight'),
+    ('position_embedding/embeddings', 'token_pos_embed._embed.weight'),
+    ('embeddings/layer_norm/', 'embed_layer_norm'),
     ('embedding_projection', 'embed_factorized_proj'),
     ('self_attention/attention_output', 'attention_proj'),
     ('self_attention_layer_norm', 'layer_norm'),