lucidrains / iTransformer

Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks, out of Tsinghua / Ant group
MIT License
445 stars 36 forks source link

ModuleNotFoundError: No module named 'iTransformer.iTransformerNormConditioned' #22

Closed Mikivishy closed 9 months ago

Mikivishy commented 9 months ago

(shy_py38) zhenningli@iotsc-zhenningli:~/shy$ /home/zhenningli/anaconda3/envs/shy_py38/bin/python /home/zhenningli/shy/DSTA2/test.py Traceback (most recent call last): File "/home/zhenningli/shy/DSTA2/test.py", line 2, in import iTransformer File "/home/zhenningli/anaconda3/envs/shy_py38/lib/python3.8/site-packages/iTransformer/init.py", line 10, in from iTransformer.iTransformerNormConditioned import iTransformerNormConditioned ModuleNotFoundError: No module named 'iTransformer.iTransformerNormConditioned'

I installed iTransformer using 'pip install Transformers' according to the readme. But when I run the example, I get an error and I cannot import iTransformer correctly. My torch version is 2.1.0 and CUDA version is 11.8. Looking forward to your answer, thank you very much.

lucidrains commented 9 months ago

@mkkk1112 oops, thanks for reporting!

Mikivishy commented 9 months ago

@mkkk1112 oops, thanks for reporting!

When I comment out the following code in init.py, the error disappears and it seems like it has no effect on normal operation,thanks for your reply.

from iTransformer.iTransformerNormConditioned import iTransformerNormConditioned