使用pytorch进行训练,不熟悉使用方法。训练时候采用了两块GPU,测试时候就报以上错误,查看了有关博客,找到原因:训练时候若设置使用多GPU训练,如下图:
本来如果单GPU的话,没有中间那一句,那一句是我后来加上的。
那么在测试的时候,也应该加上dataparallel这句话,否则就报以上错误:
如图,加了net=torch.nn.DataParallel(net)这句话,可以运行。
使用pytorch进行训练,不熟悉使用方法。训练时候采用了两块GPU,测试时候就报以上错误,查看了有关博客,找到原因:训练时候若设置使用多GPU训练,如下图:
本来如果单GPU的话,没有中间那一句,那一句是我后来加上的。
那么在测试的时候,也应该加上dataparallel这句话,否则就报以上错误:
如图,加了net=torch.nn.DataParallel(net)这句话,可以运行。