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

I replaced the individual conv in yolo with odconv and the following error occurred #27

Closed
trytry1206 opened this issue Feb 27, 2023 · 5 comments

Comments

@trytry1206
Copy link

image
I replaced the individual conv in yolo with odconv and the following error occurred
image

@trytry1206
Copy link
Author

而且我yolo中设置的batch size并不是1

@ZVChen
Copy link

ZVChen commented Mar 1, 2023

yolo在开始训练前会尝试输入一个随机张量来测试模型,可以通过判断来跳过BN层。批量为1的情况下,BN层实际上是不起作用的

@CVwbp
Copy link

CVwbp commented Mar 1, 2023

yolo在开始训练前会尝试输入一个随机张量来测试模型,可以通过判断来跳过BN层。批量为1的情况下,BN层实际上是不起作用的

请问您有没有试过加载作者提出的预训练权重呢?

@ZVChen
Copy link

ZVChen commented Mar 1, 2023

yolo在开始训练前会尝试输入一个随机张量来测试模型,可以通过判断来跳过BN层。批量为1的情况下,BN层实际上是不起作用的

请问您有没有试过加载作者提出的预训练权重呢?

没有使用这个预训练权重,只是尝试替换了yolo中的卷积

@trytry1206
Copy link
Author

yolo在开始训练前会尝试输入一个随机张量来测试模型,可以通过判断来跳过BN层。批量为1的情况下,BN层实际上是不起作用的

感谢 我之前是删去了foward里面的BN层 可以正常运行 我现在试试通过判断来跳过BN层

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