Skip to content

GitLab

Commit f851fb55 authored by thomwolf's avatar thomwolf
Browse files

fixing error message

parent 50c6bc41
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_tf_bert.py
View file @ f851fb55
...@@ -113,7 +113,7 @@ def load_bert_pt_weights_in_tf2(tf_model, config, pytorch_checkpoint_path): ...@@ -113,7 +113,7 @@ def load_bert_pt_weights_in_tf2(tf_model, config, pytorch_checkpoint_path):
tfo = tf_model(tf_inputs, training=False) # Make sure restore ops are run tfo = tf_model(tf_inputs, training=False) # Make sure restore ops are run
logger.info("Weights not loaded: {}".format(all_pytorch_weights)) logger.info("Weights or buffers not loaded from PyTorch model: {}".format(all_pytorch_weights))
return tf_model return tf_model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment