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.

Sign up for GitHub

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

Jump to bottom

关于使用SwinTransformer的GFLOPs疑问 #80

Open
LUO77123 opened this issue May 9, 2022 · 1 comment
Open

关于使用SwinTransformer的GFLOPs疑问 #80

LUO77123 opened this issue May 9, 2022 · 1 comment

Comments

@LUO77123
Copy link

LUO77123 commented May 9, 2022

大佬您好，我自己根据SwinTransformer源码将YOLOV5的6.1版本的backbone全部按SwinTransformer结构修改，遇到的半精度训练报错问题看您的train文件添加了'--nohalf'参数后成功了，但是我的GFLOPs很大，光depth_multiple: 0.33，width_multiple: 0.75的yolov5swin_tiny就409.9GFLOPs，而depth_multiple: 1，width_multiple:1的yolov5swin_base达到离谱的1307GFLOPs，就算导入并冻结预训练权重也训练很慢。大佬是否遇到GFLOPs过大的问题，过大是否需要剪枝（还没接触过剪枝，请大佬赐教。感谢
@lost0fate
Copy link

您好，我想请问一下你后面解决这个问题了嘛？我是在最后的3个探测头上面，把C3TR换为了C3STR，结果增加了200多GFLOPs。后面测试只在第一个探测头（P3）上更改的话，仅增加个位数的GFLOPs，但如果只给最后一个探测头（P5）更换，则会陡增。想请教下您最后是怎么解决这个问题的或是告知下这个问题产生的原因

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@LUO77123 @lost0fate