[Python] 不要看到No module named ‘torch._six‘ 就慌忙升级或降级Pytorch版本，不妨尝试改一点代码。

不要看到No module named 'torch._six' 就慌忙升级或降级Pytorch。不妨尝试改一点代码。

实测中，发现用到torch._six的，往往是跟一些数据类型或者数学有关的。

那么以下有些情况，可以直接改代码。

情况1：

from torch._six import inf
...
if norm_type == inf:
    do_something()

这种情况完全可以用math.inf来代替，修改如下

import math
#from torch._six import inf
...
if norm_type == math.inf:
    do_something()

情况2：

import torch._six

...

if isinstance( value, torch._six.string_classes) :
   do_something()

这种情况完全可以用str来代替，修改如下

#import torch._six

...

if isinstance( value, str) :
   do_something()

同样，如果出现 torch._six.int_classess，可以用int来代替。

具体情况具体分析，不要着急听信网上那些动不动就升级或降低Pytorch版本什么的。