search

649453932 / Bert-Chinese-Text-Classification-Pytorch

使用Bert，ERNIE，进行中文文本分类
MIT License
3.97k stars 897 forks source link

添加 poetry 的 pyproject.toml 文件。支持了 apple mps GPU 加速训练。 #195

Open GuoDapeng opened 2 months ago

GuoDapeng commented 2 months ago

训练中

截屏2024-06-28 15 51 12

训练结束

截屏2024-06-28 16 15 43

目测效果不错

GuoDapeng commented 2 months ago

在 128 GB 内存的 macOS ，self.pad_size 可以设置到 512，虽然不知道有什么用，但是能在 macOS 上训练解决了我台式机显存不足的问题，