![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
imaginist233
这个作者很懒,什么都没留下…
展开
-
简单手动实现pytorch中的MaxPooling层
简单手动实现pytorch中的MaxPooling层本来想去看一下pytorch中的MaxPooling层的源码,结果没有找到具体过程,于是自己实现了一下,代码如下。import torchimport torch.nn as nnclass MyMaxPool2D(nn.Module): def __init__(self, kernel_size=(2, 2), strid...原创 2020-04-27 21:02:29 · 6008 阅读 · 0 评论 -
RuntimeError: unexpected EOF. The file might be corrupted.
最近在使用pytorch加载训练好的模型时出现了这个问题,最后发现是通过filezilla传输过来的模型损坏了(文件不完整),重新传输一次就好了。原创 2019-03-05 10:38:48 · 8297 阅读 · 6 评论 -
Pytorch GPU显存充足却显示out of memory解决办法
今天在测试一个pytorch代码的时候显示显存不足,但是这个网络框架明明很简单,用CPU跑起来都没有问题,GPU却一直提示out of memory.在网上找了很多方法都行不通,最后我想也许是pytorch版本的问题,原来我的pytorch版本是0.4.1,于是我就把这个版本卸载,然后安装了pytorch1.1.0,程序就可以神奇的运行了,不会再有OOM的提示了。虽然具体原因还不知道为何,这里还...原创 2019-06-13 14:20:40 · 26188 阅读 · 13 评论