1 Star 0 Fork 2

小码编程AI / bert_train

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
run.sh 600 Bytes
一键复制 编辑 原始数据 按行查看 历史
lee 提交于 2019-04-21 17:04 . 人肉调参
export BERT_BASE_DIR=G:/bert-use-demo/chinese_L-12_H-768_A-12
export MY_DATASET=G:/bert-use-demo/data
python bert/run_classifier.py \
--data_dir=G:/bert-use-demo/data \
--task_name=sim \
--vocab_file=G:/bert-use-demo/chinese_L-12_H-768_A-12/vocab.txt \
--bert_config_file=G:/bert-use-demo/chinese_L-12_H-768_A-12/bert_config.json \
--output_dir=/tmp/sim_model/ \
--do_train=true \
--do_eval=true \
--init_checkpoint=G:/bert-use-demo/chinese_L-12_H-768_A-12/bert_model.ckpt \
--max_seq_length=128 \
--train_batch_size=32 \
--learning_rate=5e-5\
--num_train_epochs=2.0
PAUSE
Python
1
https://gitee.com/small_code_programming_ai/bert_train.git
git@gitee.com:small_code_programming_ai/bert_train.git
small_code_programming_ai
bert_train
bert_train
master

搜索帮助