做神经网络的一个报错
原因:输入数据和参数的数据格式不匹配,应该是数据传到了GPU上而,神经网络模型没有传输到GPU上。
解决:在实例化网络后,加一条传到GPU上的代码即可。
model=DANN() # DANN是你自己搭建的神经网络,实例化神经网络
device = torch.device("cuda:" + config.gpu_id if torch.cuda.is_available() else "cpu")
model=model.to(device) #传到GPU
做神经网络的一个报错
原因:输入数据和参数的数据格式不匹配,应该是数据传到了GPU上而,神经网络模型没有传输到GPU上。
解决:在实例化网络后,加一条传到GPU上的代码即可。
model=DANN() # DANN是你自己搭建的神经网络,实例化神经网络
device = torch.device("cuda:" + config.gpu_id if torch.cuda.is_available() else "cpu")
model=model.to(device) #传到GPU