UnitEnglisharoundtheworld词汇图文实用教案.pptx
2021-12-14 09:03:30
1.14MB
pytorch实现的 + 新闻多分类样例
1.安装依赖
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
2.WoBERT模型下载
方式1.
从 下载keras模型，然后执行转换变成pytorch的模型
python config/conver.py
方式2.
直接下载使用 ， 链接: 密码: 94au
3.配置参数
vim config/configs.yaml
4.train\test
python train.py
5.案例数据
data/train.csv
细节
除了transformers转换模型，主要是tokenizer的变化
1.稍微自定义了个专门的分词，见src/models/bert_model.py:WoBertTokenizer
2.稍微自定义了个专门的BERT分
2021-12-13 17:26:22
21KB
1