老师请问我显卡1660ti 6G版的batchSize和epoch选多少合适？

来源：8-25 训练脚本编写及演示

YaphetSNice3502248

2020-03-31

老师请问我显卡1660ti 6G版的batchSize和epoch选多少合适？按照你们batchSize=1024和epoch=1000会出现显存爆炸的情况，还会出现cuDNN error，这该如何处理

写回答

2回答

胖虎

2020-07-22

OOM 是因为你的GPU显存不够导致的，建议调小BatchSize 默认1024 可以改到512 256 试试

胖虎

2020-03-31

调小batchsize

基于Pytorch热门深度学习框架从零开发NLP聊天机器人

基于中科院所内项目，Seq2Seq+attention主流算法，实战聊天机器人

273 学习 · 107 问题

相似问题

回答 3

回答 3

回答 1

回答 1

回答 1