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

如何修改batch_size #19

Closed

tongyang-218 opened this issue Mar 15, 2023 · 1 comment

tongyang-218 commented Mar 15, 2023 •

edited

Loading

张博士您好，代码中修改batch_size的地方应该在哪里，配置文件里面没有这一处
在运行代mf模块的train文件中报错ZeroDivisionError: float division by zero

这两个地方的代码怎么理解，ch_in传递变成了64-3-1

tongyang-218 changed the title ~~如何修改btach_size~~ 如何修改batch_size

Owner

icey-zhang commented Mar 20, 2023

修改batch size：

SuperYOLO/train.py

Line 589 in c6aca72

    
           parser.add_argument('--batch-size', type=int, default=2, help='total batch size for all GPUs')

这个结构在论文终版中有提到，分别对应这张图上的三个seblock
https://github.com/icey-zhang/SuperYOLO/blob/main/Fig/Fusion_se.png

icey-zhang closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment