-
Notifications
You must be signed in to change notification settings - Fork 700
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
resnet50 + ghost module #34
Comments
resnet-50 中的bottleneck 里的 3个 conv 都换成ghost module吗 |
是的,除了第一个conv,其他所有conv都换。 |
You mean, use Ghost_module instead of all the conv in the bottleneck ,include conv in downsample and except the first conv in Resnet? So I replace all of the conv except the first conv in Resnet, but The GPU cost increasely almost 10897M( My decive: 1 piece 1080ti GPU, before modify, GPU cost is 7645M ,batchsize= 8 * 4 ),it remind me that GPU is run out of memory, until I modify the batchsize = 1*4, the program can run! |
@iamhankai @huangpan2507 |
您好,要传进去的。primary_conv接收这2个参数,cheap_operation不接收。 |
不好意思,我按照您的意思进行更改。发现GhostModule应用在stride=2的conv上会出现,和原始conv得到的特征图维度不一致的情况。如:输入6336的特征图,stride=2的conv得到3131的输出特征图,而ghostmodule得到的是32*32输出特征图。这样会影响网络的输出与训练标签不一致,请问一下,您们是怎么解决的啊?谢谢 |
你需要改变conv或cheap_op的padding值,以让他们的输出尺寸能匹配上。 |
感谢作者的耐心回复!在您的指点下,成功替换了resnet50中的conv(除了第一个conv和一个stride=2的conv)。但当我将替换后的backobone进行训练时,会报错: |
推测是因为中间过程的feature maps更多了。减小batch size吧。 |
ghostnet在 resnet50中的那个实验,是怎么把ghost module结合进去的呀,cheap conv用的是depth conv吗
The text was updated successfully, but these errors were encountered: