- 博客(2)
- 资源 (6)
- 收藏
- 关注
RSS订阅原创 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
1248
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
1485
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人