RuntimeError: expected scalar type Double but found Float

最近在使用 sequitur库 快速搭建自编码器时遇到 RuntimeError: expected scalar type Double but found Float

涉及代码

import torch
from sequitur.models import LINEAR_AE

model = LINEAR_AE(
  input_dim=300,
  encoding_dim=20,
  h_dims=[120, 60],
  h_activ=None,
  out_activ=None
)

model

错误的意思是期待的张量类型是double但是输入的是float（有时候这个提示不一定是准确的，但归根就是类型错误），可以将所有的层的输入输出类型打印出来：

for name, param in model.named_parameters():
    print(name,'-->',param.type(),'-->',param.dtype,'-->',param.shape)

学习

发现全部都是tensor.float32 类型，再检查一下自己的输入数据类型，发现不对，使用类型转换 valset = torch.tensor(valset,dtype=torch.float) 即可。

另外torch 查看模型参数

# 构建模型
class Net(nn.Module):

# 实例化模型
net = Net()

# 查看模型的各层的尺寸
for name,parameters in net.named_parameters():
    print(name,':',parameters.size())