新写了个TextRNN跑文本分类
pytorch吧
全部回复
仅看楼主
level 3
◎御坂雷丘◎
楼主
但是莫名其妙的就会突然申请很多内存然后直接炸掉，训练小点的数据集设置batch为32几百步就会这直接卡死，但是大点的数据集就可以训练多一些。
这有可能是什么原因造成的？
2023年04月06日 15点04分
1
level 11
sealed_ss
rnn在某些cuda上兼容性不好，文档里写了
2023年04月07日 03点04分
2
◎御坂雷丘◎
我已经在大的数据集上训练完了，一点问题都没有，跑到后面我的内存总占用甚至还在跌。
2023年04月07日 03点04分
◎御坂雷丘◎
@sealed_ss
而且在小的数据集上无论开不开cuda都会疯狂申请直到容量不够，然后炸掉，吓得我训练的batch只敢开16
2023年04月07日 03点04分
sealed_ss
@◎御坂雷丘◎
对啊对啊是这样的，我换电脑了之后就没这问题了
2023年04月07日 04点04分
1
