transformers-cli 错误：需要以下参数：--model_type

Question

我正在尝试使用transformers-cli 将 tf 检查点转换为 pytorch 检查点，如下所示

transformers-cli convert model_type bert --tf_checkpoint bio_bert_large_1000k.ckpt --config bert_config_bio_58k_large.json --pytorch_dump_output pytorch_model.bin

我收到以下错误

usage: transformers-cli <command> [<args>] convert [-h] --model_type
                                               MODEL_TYPE --tf_checkpoint
                                               TF_CHECKPOINT
                                               --pytorch_dump_output
                                               PYTORCH_DUMP_OUTPUT
                                               [--config CONFIG]
                                               [--finetuning_task_name FINETUNING_TASK_NAME]
transformers-cli <command> [<args>] convert: error: the following arguments are required: --model_type

我做错了什么？

Answer 1

发现错误..它必须是--model_type，而不是model_type。