Apperently you need to type case (or initialize) one of your variables to another, shorter data type.
As the error indicates, try casting it torch.float32:
image1091×423 22.7 KB
It’s a bit difficult to follow your code example, so I can’t really tell what variable causes the problem.