由于之前用python2以及pytorch0.2.0跑代码load模型时太慢了，于是换成了python3 并将pytorch升级为0.4.1。load模型快多了。
注意代码中需要更改的地方：
1. python2中 / 可以表示整除，当除数和被除数都是整数时。在python3中都要改成 //
2. backward()的参数retain_variables=True 在pytorch新版本中改为了 retain_graph=True
python2 和 python3的区别，以及pytorch升级
原文：https://www.cnblogs.com/amarr/p/10404593.html