Bert-Chinese-Text-Classification-Pytorch
Bert-Chinese-Text-Classification-Pytorch copied to clipboard
添加 poetry 的 pyproject.toml 文件。支持了 apple mps GPU 加速训练。
训练中
训练结束
目测效果不错
在 128 GB 内存的 macOS ,self.pad_size 可以设置到 512,虽然不知道有什么用,但是能在 macOS 上训练解决了我台式机显存不足的问题,