WARNING:root:You are using an old version PyTorch 0.1.12, which THOP is not going to support in the future.
Traceback (most recent call last):
File "/home/lzs/lzs/TuSimple/LSTR/test.py", line 14, in
from nnet.py_factory import NetworkFactory
File "/home/lzs/lzs/TuSimple/LSTR/nnet/py_factory.py", line 5, in
from thop import profile, clever_format
File "/home/lzs/anaconda3/envs/TuSimple/lib/python3.6/site-packages/thop/init.py", line 2, in
from .profile import profile, profile_origin
File "/home/lzs/anaconda3/envs/TuSimple/lib/python3.6/site-packages/thop/profile.py", line 23, in
default_dtype = torch.float64
**AttributeError: module 'torch' has no attribute 'float64'
**
Process finished with exit code 1
（pytorch版本就是这个，没有新版本的）
升级下PyTorch