RuntimeError: expected scalar type Byte but found Float

本文主要是介绍RuntimeError: expected scalar type Byte but found Float，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

部分代码如下

for data in train_loader:imgs, targets = data# print("标签为", targets.shape)imgs.byte()output = unet_l2(imgs)loss = criterion(output, targets)optimizer.zero_grad()loss.backward()optimizer.step()

报错如下：

RuntimeError: expected scalar type Byte but found Float

解决方法

output = unet_l2(imgs.to(torch.float32))

这篇关于RuntimeError: expected scalar type Byte but found Float的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---