RuntimeError: "mse_cuda" not implemented for 'Long' 报错

criterion = nn.MSELoss()
criterion(a, b)
这是a的dtype=torch.float，b的dtype=torch.int64
因此，都改成float
 
posted @ 2021-11-07 22:23  Rogn  阅读(6942)  评论(0)  编辑  收藏  举报