Open sparkydogX opened 5 years ago
https://sparkydogx.github.io/2018/08/08/pytorch-dataparallel-problem/
pytorch中如果使用DataParallel，那么保存的模型key值前面会多处’modules.’，这样如果训练的时候使用的是多GPU，而测试的时候使用的是单GPU，模型载入就会出现问题。一个解决方法是测试的时候强行DataParallel，但是有时候情况较为复杂，可以使用如下的方法：（参考来源：https://discuss.pytorch.org/t/solved-keyerror-une
https://sparkydogx.github.io/2018/08/08/pytorch-dataparallel-problem/
pytorch中如果使用DataParallel，那么保存的模型key值前面会多处’modules.’，这样如果训练的时候使用的是多GPU，而测试的时候使用的是单GPU，模型载入就会出现问题。一个解决方法是测试的时候强行DataParallel，但是有时候情况较为复杂，可以使用如下的方法：（参考来源：https://discuss.pytorch.org/t/solved-keyerror-une