microsoft / CSWin-Transformer

CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022
MIT License
539 stars 78 forks source link

timm version #49

Open haveasip-and-code opened 7 months ago

haveasip-and-code commented 7 months ago

I was trying to train this model on GG Colab. At first, I installed the latest version of timm (0.9.12) and I encountered this error: cannot import name 'Dataset' from 'timm.data'

Then I reinstalled the timm version that was recommended in the readme.md file (0.3.4) and I saw another error: File "/root/.local/lib/python3.10/site-packages/timm/models/layers/helpers.py", line 6, in <module> from torch._six import container_abcs ModuleNotFoundError: No module named 'torch._six'

Can you please suggest any ways to fix this? Thank you so much.

niceyh888 commented 5 months ago

I have the same question