mindspore-lab / mindcv

A toolbox of vision models and algorithms based on MindSpore
https://mindspore-lab.github.io/mindcv/
Apache License 2.0
230 stars 139 forks source link

fix: Use custom GroupConv instead of nn.Conv2d in shufflenet_v1 #763

Closed The-truthh closed 5 months ago

The-truthh commented 5 months ago

Thank you for your contribution to the MindCV repo. Before submitting this PR, please make sure:

Motivation

Due to MindSpore doesn't support group conv in shufflenet, we need to define the group convolution manually, instead of using the origin nn.Conv2d by changing the parameter group.

Test Plan

(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)

Related Issues and PRs

(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)