报错:
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
检查发现我的input和weight都是float32,所以不是我的类型发生了错误;
最后发现是我的网络 没有加载到gpu里面。
model = UNet().cuda()
添加cuda即可。
报错:
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
检查发现我的input和weight都是float32,所以不是我的类型发生了错误;
最后发现是我的网络 没有加载到gpu里面。
model = UNet().cuda()
添加cuda即可。