2018年11月5日
pytorch 在测试阶段，进行前向推断时，显存累加溢出问题
摘要： 1、问题描述： pytorch中，在测试阶段进行前向推断运行时，随着for循环次数的增加，显存不断累加变大，最终导致显存溢出。 2、解决方法： 使用如下代码处理输入数据： 假设X为模型的输入 X = X.cuda() input_blobs = Variable(X, volatile=True) 阅读全文
posted @ 2018-11-05 18:13 blackx 阅读(4429) 评论(0) 推荐(0) 编辑