update to hf->tf args

8de1faea · Chris · d0adab2c · 8de1faea
Commit 8de1faea authored May 22, 2019 by Chris
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/convert_pytorch_checkpoint_to_tf.py pytorch_pretrained_bert/convert_pytorch_checkpoint_to_tf.py +1 -1

No files found.
--- a/pytorch_pretrained_bert/convert_pytorch_checkpoint_to_tf.py
+++ b/pytorch_pretrained_bert/convert_pytorch_checkpoint_to_tf.py
@@ -22,7 +22,7 @@ import tensorflow as tf
 from pytorch_pretrained_bert.modeling import BertConfig, BertModel


-def convert_pytorch_checkpoint_to_tf(model:type(BertModel), ckpt_dir:str):
+def convert_pytorch_checkpoint_to_tf(model:BertModel, ckpt_dir:str):

    """
    :param model:BertModel Pytorch model instance to be converted