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。