search

yuanzhoulvpi2017 / zero_nlp

中文nlp解决方案(大模型、数据、模型、训练、推理)
MIT License
2.85k stars 355 forks source link

code02_训练模型全部流程.ipynb运行问题 #100

Closed situjunhao closed 1 year ago

situjunhao commented 1 year ago

data_collator函数中的eop_token_id报错，是eos_token_id吗？ 其次，爆显存，本人机器16G显存，请问需要多少显存呢？