ModuleNotFoundError: No module named 'timm'

[Logan-lu]

Following setup steps in README.md, when running training script (python train.py --data coco128.yaml --cfg configs/yolov5/yolov5s.yaml), met the following errors:

Downloading https://ultralytics.com/assets/Arial.ttf to /home/logan/.config/Ultralytics/Arial.ttf...
Traceback (most recent call last):
File "/home/logan/dev/pyfarm/pytorch/yoloair/train.py", line 41, in
import val # for end-of-epoch mAP
File "/home/logan/dev/pyfarm/pytorch/yoloair/val.py", line 38, in
from models.common import DetectMultiBackend
File "/home/logan/dev/pyfarm/pytorch/yoloair/models/common.py", line 35, in
from models.Models.muitlbackbone import conv_bn_hswish, MobileNetV3_InvertedResidual, DepthSepConv,
File "/home/logan/dev/pyfarm/pytorch/yoloair/models/Models/muitlbackbone.py", line 26, in
from models.Models.SwinTransformer import SwinTransformerLayer
File "/home/logan/dev/pyfarm/pytorch/yoloair/models/Models/SwinTransformer.py", line 9, in
from timm.models.layers import DropPath, to_2tuple, trunc_normal_
ModuleNotFoundError: No module named 'timm'

Any help?

P.S. My env: Ubuntu22.04.1, python 3.10.4

[iscyy]

you can use pip install timm

[zhongsir771]

pip is useful