ImportError running detr

[LSC527]

🐛 Describe the bug

File "/workspace/ColossalAI-Examples/image/detr/models/transformer.py", line 10, in
from titans.layer.attention import DeTrAttention
ImportError: cannot import name 'DeTrAttention' from 'titans.layer.attention' (/opt/conda/lib/python3.8/site-packages/titans/layer/attention/init.py)

Environment

No response

[binmakeswell]

Hi @LSC527 Thank you for your feedback. We will try to reproduce your issue.
This is another version DeTr
https://github.com/hpcaitech/ColossalAI-Examples/tree/main/image/detr-debug

By the way, we are restructuring the documents and examples, and the new version examples will be provided at the following link
https://github.com/hpcaitech/ColossalAI/tree/main/examples