![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
nlite827109223
这个作者很懒,什么都没留下…
展开
-
pytorch 模型不在类的 __init__()函数中定义,且该函数中无可学习的参数
如这种形式,反卷积方法在conv1d_transpose中定义,在训练的时候,model.parameters()会为空,在__init__()中加一部分带参数的方法即可; 报的错为:raise ValueError("optimizer got an empty parameter list") ValueError: optimizer got an empty parameter l...原创 2019-12-11 19:48:38 · 1232 阅读 · 2 评论 -
记录pytorch floattensor与cuda.floattensor的类型不一致的问题
1. 一般报这个错,是因为使用gpu,但是数据类型为转换,使用data.to(device)即可, device是device = torch.device("cuda:0" if cuda_used==1 and torch.cuda.is_available() else 'cpu') 2. 我这里是因为定义模型的时候,有部分模型定义在 def __init__(self)的外部,虽然使用...原创 2019-12-01 20:12:05 · 1475 阅读 · 0 评论 -
pytorch 测试时候显存溢出
cuda out of memory 解决方法:http://www.cnblogs.com/Charlene-HRI/p/10234656.html 我的问题时在测试时候需要保存每次的输出中间向量,忘记转为 cpu().numpy(), 导致显存累积,越来越大,最终溢出 ...转载 2019-05-05 15:38:49 · 4303 阅读 · 0 评论 -
pytorch 反卷积
1. condtranspose2d import torch from torch import nn from torch.nn import init from torch.autograd import Variable import torchvision dconv = nn.ConvTranspose2d(in_channels=1, out_channels= 1, kern...转载 2019-05-30 11:35:35 · 895 阅读 · 0 评论