[X] I have searched the YOLOAir issues and found no similar bug report.
YOLOAir Component
No response
Bug
(base) ➜ yoloair git:(main) ✗ python train.py
Traceback (most recent call last):
File "train.py", line 41, in
import val # for end-of-epoch mAP
File "/home/featurize/work/yoloair/val.py", line 38, in
from models.common import DetectMultiBackend
File "/home/featurize/work/yoloair/models/common.py", line 36, in
from models.Models.muitlbackbone import conv_bn_hswish, DropPath, MobileNetV3_InvertedResidual, DepthSepConv, \
File "/home/featurize/work/yoloair/models/Models/muitlbackbone.py", line 26, in
from models.Models.SwinTransformer import SwinTransformerLayer
File "/home/featurize/work/yoloair/models/Models/SwinTransformer.py", line 9, in
from timm.models.layers import DropPath, to_2tuple, truncnormal
ModuleNotFoundError: No module named 'timm'
Hyper-Devil
commented
2 years ago
Search before asking
YOLOAir Component
No response
Bug
(base) ➜ yoloair git:(main) ✗ python train.py
import val # for end-of-epoch mAP
File "/home/featurize/work/yoloair/val.py", line 38, in
from models.common import DetectMultiBackend
File "/home/featurize/work/yoloair/models/common.py", line 36, in
from models.Models.muitlbackbone import conv_bn_hswish, DropPath, MobileNetV3_InvertedResidual, DepthSepConv, \
File "/home/featurize/work/yoloair/models/Models/muitlbackbone.py", line 26, in
from models.Models.SwinTransformer import SwinTransformerLayer
File "/home/featurize/work/yoloair/models/Models/SwinTransformer.py", line 9, in
from timm.models.layers import DropPath, to_2tuple, truncnormal
ModuleNotFoundError: No module named 'timm'
Traceback (most recent call last): File "train.py", line 41, in
Environment
No response
Minimal Reproducible Example
No response
Additional
No response
Are you willing to submit a PR?