search

QY1994-0919 / CFPNet

Centralized Feature Pyramid for Object Detection
Apache License 2.0
237 stars 22 forks source link

显存会随轮数无限变大吗 #19

Open DirtyBit64 opened 1 year ago

DirtyBit64 commented 1 year ago

我训到70轮显存溢出了，总共还需要训400轮,,,,

dongzhang89 commented 1 year ago

@PGthree3 不会，猜测是你代码的问题

DirtyBit64 commented 1 year ago

我降低了两个bs，并且将block的中间扩张率下调成了2，有一定涨点效果，请问作者有试过调整EVCblock的中间扩张率吗，源码是4和2差距大吗

dongzhang89 commented 1 year ago

@PGthree3 扩张率是一个超参数，应该在不同数据集上性能有所差距。我们当时没有做这么多的调参，EVCblock只是提供了一种除了attention之外的其他可能性。要是你有更多细节的实验结果，欢迎report或者提交pull request，谢谢。

DirtyBit64 commented 1 year ago

好的，感谢回复！

SmoothJing commented 1 year ago

我降低了两个bs，并且将block的中间扩张率下调成了2，有一定涨点效果，请问作者有试过调整EVCblock的中间扩张率吗，源码是4和2差距大吗

请问你是直接用的CFP模块还是只用了EVC模块啊