code02_训练模型全部流程.ipynb运行问题 - Githubissues

yuanzhoulvpi2017 / zero_nlp

中文nlp解决方案(大模型、数据、模型、训练、推理)

MIT License

3.04k stars 369 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显存，请问需要多少显存呢？