Comments (8)
xu-cpp
commented on July 20, 2024
2
试试在run.py里加上这两行代码
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2, 3'
from bert-chinese-text-classification-pytorch.
MambaMelon
commented on July 20, 2024
还想请教一下，你的博客地址在哪里啊？可否贴一下
from bert-chinese-text-classification-pytorch.
dabao0903
commented on July 20, 2024
同样的问题，执行bert 修改了pad_size，报错CUDA out of memory，好像是模型导入pretrained时候，出错了，有遇到的朋友请解答下疑惑，谢谢~
from bert-chinese-text-classification-pytorch.
moluchase
commented on July 20, 2024
同样的问题，执行bert 修改了pad_size，报错CUDA out of memory，好像是模型导入pretrained时候，出错了，有遇到的朋友请解答下疑惑，谢谢~
请问，该问题解决了吗
from bert-chinese-text-classification-pytorch.
wangjs9
commented on July 20, 2024
可以更改config中batch_size的
from bert-chinese-text-classification-pytorch.
dabao0903
commented on July 20, 2024
可以更改config中batch_size的
应该不是这个问题，试过batch_size=1，也会报错
from bert-chinese-text-classification-pytorch.
moluchase
commented on July 20, 2024
可以更改config中batch_size的
应该不是这个问题，试过batch_size=1，也会报错
是这个问题，我已经试过了，修改batch_size或者修改pad_size
from bert-chinese-text-classification-pytorch.
limo1995
commented on July 20, 2024
你显存不够吧
from bert-chinese-text-classification-pytorch.
Related Issues (20)
- 怎么测试自己的数据 HOT 1
- Question about masked language model
- 如何改繁體訓練
- 训练的准确率过低 HOT 2
- python: can't open file 'D:\Bert-Chinese-Text-Classification-Pytorch-master\run.py': [Errno 2] No such file or directory
- 字向量维度是768吗
- 请问一下，模型跑出来的结果，只有准确率 丢失率等等这些吗？具体的每个句子的用bert进行分类的标签，在哪里可以看到呀？ HOT 7
- OSError: Can't load the model for 'D:\Bert-Chinese-Text-Classification-Pytorch-master\THUCNews\saved_dicert.ckpt'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'D:\Bert-Chinese-Text-Classification-Pytorch-master\THUCNews\saved_dicert.ckpt' is the correct path to a di rectory containing a file named pytorch_model.bin, tf_model.h5, model.ckpt or flax_model.msgpack.
- 模型准确率 HOT 1
- 已经微调好的模型调用不了怎么办？只生成一个bert.ckpt文件
- 为什么预测的结果不太准呀？而且每预测一次，得到的分类结果不一样 HOT 6
- 我加载了微调之后的权重模型bert.ckpt用他来进行短文本预测 但是每次运行文件时的预测结果都会变化且预测效果不好 应该怎么解决呀？
- 请问在调用模型预测时 出现这个报错是为什么？且预测的结果不太准，跟这个报错有关系吗
- 请问有人使用ernie在自己数据集上做训练的时候loss无法下降，准确率上不来嘛？同样的数据bert很正常
- 词表下载链接好像不太对，无法下载 HOT 4
- 能否通过调接口的方式来进行训练，每次启动都要训练好久
- 小白想问下能否用这个代码训练Roberta模型
- 求大佬解答，按照词分割，具体怎么改
- 词表txt文件 HOT 2
- 长文本
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from bert-chinese-text-classification-pytorch.