Colab来训练bert模型

本文主要是介绍Colab来训练bert模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Colab使用

1.切换tensorflow版本号

%tensorflow_version 1.x

2.查看tensorflow版本

!pip show tensorflow

3.训练模型

!python /content/drive/My\ Drive/bert/run_english_classify.py --task_name=mypro  --do_train=true --do_eval=true  --data_dir=/content/drive/My\ Drive/english --vocab_file=/content/drive/My\ Drive/wwm_uncased_L-24_H-1024_A-16/vocab.txt --bert_config_file=/content/drive/My\ Drive/wwm_uncased_L-24_H-1024_A-16/bert_config.json --num_train_epochs=100 --train_batch_size=5 --max_seq_length=50 --output_dir=/content/drive/My\ Drive/english_model/fine-tuning-model

4.导出模型

!python /content/drive/My\ Drive/bert/run_english_classify.py --task_name=mypro --do_predict=true --data_dir=/content/drive/My\ Drive/english --vocab_file=/content/drive/My\ Drive/wwm_uncased_L-24_H-1024_A-16/vocab.txt --bert_config_file=/content/driv

这篇关于Colab来训练bert模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---