diff --git a/baselines/models_pytorch/mrc_pytorch/convert_tf_checkpoint_to_pytorch.py b/baselines/models_pytorch/mrc_pytorch/convert_tf_checkpoint_to_pytorch.py index 4ed8529..5eb0234 100755 --- a/baselines/models_pytorch/mrc_pytorch/convert_tf_checkpoint_to_pytorch.py +++ b/baselines/models_pytorch/mrc_pytorch/convert_tf_checkpoint_to_pytorch.py @@ -116,7 +116,6 @@ def convert_tf_checkpoint_to_pytorch(tf_checkpoint_path, bert_config_file, pytor parser.add_argument("--is_albert", default=False, action='store_true', - type=bool, help="whether is albert?") args = parser.parse_args() convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path,