pytorch
万万VV
想成为技术大牛的软件工程专业菜鸡学生一枚......
展开
-
[pytorch]亲测解决RuntimeError: CUDA out of memory
问题当我在测试训练好的基于Pytorch框架的半监督视频目标分割模型时,我已经加上了Model.eval( ),用于测试,但是运行过程中报错:RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.95 GiB total capacity; 3.39 GiB already allocated; 9.88...原创 2020-03-17 17:47:25 · 12667 阅读 · 3 评论 -
[pytorch笔记]net.eval和with torch_no_grad
在使用net.eval( )和with torch_no_grad( )时,发现网上有的例子只使用了一个,有些两个都用了,我之前也认为这两个的作用一样,后来发现最好是两个都用。with torch_no_grad( ):数据不需要梯度计算,即不会进行反相传播;net.eval( ):不加的话即使没有训练输入数据也会改变权值,因为这是禁止forward过程对参数造成的影响;例如禁止dropo...原创 2020-03-04 16:49:33 · 9505 阅读 · 3 评论