Bert-Chinese-Text-Classification-Pytorch 添加 poetry 的 pyproject.toml 文件。支持了 apple mps GPU 加速训练。

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

Open GuoDapeng opened this issue 1 year ago • 1 comments

训练中截屏2024-06-28 15 51 12

训练结束截屏2024-06-28 16 15 43

目测效果不错

Jun 28 '24 08:06 GuoDapeng

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

Jun 30 '24 07:06 GuoDapeng