Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training abnormally slow #40

Open
7W7W7W opened this issue Aug 18, 2024 · 2 comments
Open

Training abnormally slow #40

7W7W7W opened this issue Aug 18, 2024 · 2 comments

Comments

@7W7W7W
Copy link

7W7W7W commented Aug 18, 2024

image
Ultralytics YOLOv8.2.29 🚀 Python-3.11.9 torch-2.4.0+cu121 CUDA:0 (NVIDIA GeForce RTX 3060 Ti, 8192MiB)
Mamba-YOLO-T, It took over 30 hours, but Mamba-YOLO-L couldn't run at all. Is this normal?

@EthanW-coder
Copy link
Collaborator

Unfortunately, Mamba's training time on visual tasks is not comparable to CNN's framework, and we are working to address these issues in the future. In fact, on the COCO dataset, we used 8×H800 training, about which we elaborate in the paper, and more details will be published after the mid-draft of the paper.

@YangJae96
Copy link

Does mamba based yolo network has advantange in terms of inference speed, but not training time?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants