在对网络进行修改时,加了一个函数supervisor(),然后运行发现出现了RuntimeError: expected device cuda:0 and dtype Float but got device cpu and dtype Float 的错误
在网上搜了很多的解决方法,有说降低Torch版本的,但对我的代码并不适用,后来发现,是因为网络使用的是gpu,而supervisor()函数中的:
branch_1 = x * mask
使用的是cpu,将其改为使用gpu即可:
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
x = x.to(device)
mask = mask.to(device)
branch_1 = x * mask
再运行就成功了!