pycharm配置Bert模型运行参数

在github官网上，我们可以看到通过控制台运行run_classifier.py并加入相关训练参数可以进行模型的训练，在pycharm中，我们可以提前写好训练参数，直接点运行按钮就可以进行模型的训练，操作步骤如下：

 

1.右击项目中的run_classifier.py文件，选择Edit'run_classifer.py'

 

 2.在Parameters中写入模型训练的参数，路径使用相对路径，在写入参数时需要把github中‘\’符号去掉

 

 

 

 data_dir是模型训练和测试数据所在的文件夹

vocab.txt是词典

bert_config_file是模型的一些配置参数

init_checkpoint是预训练的模型