add code for apple silicon

Morizeyao / GPT2-Chinese

Chinese version of GPT2 training code, using BERT tokenizer.

MIT License

7.48k stars 1.7k forks source link

Open YuXHe15 opened 11 months ago

YuXHe15 commented 11 months ago

添加了Pytorch对MTS后端支持的代码，检测到完整的MTS配置后可以调用Apple Silicon的GPU进行训练

YuXHe15 commented 11 months ago

抱歉我的linter自动整理了你的代码。。。实际改动没有这么多