cywjava
cywjava
3060 12GB 设置成3 ,变成4就oom
3060 30分钟跑一轮。2M
> 我想训练一个问答模型,应该准备一些什么样子的数据呢,用这个能训练吗?
> Reference in n 我想训练一个问答模型,应该准备一些什么样子的数据呢,用这个能训练吗?
> 是的,太少了,只是尝试一下训练 。并且3060 12Gb显存 还只能设置batch_size 到3,再大就OOM
训练 时,加上--raw 会重新创建这些文件
数据集的问题。
torch的问题
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113
词表里没有你这个词,就会用UNK代替。