timm version

[haveasip-and-code]

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]

I have the same question