search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
333 stars 102 forks source link

ImportError running detr #186

Open LSC527 opened 1 year ago

LSC527 commented 1 year ago

🐛 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 commented 1 year ago

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