-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
A reimplementation of MixNet in Gluon. #900
base: master
Are you sure you want to change the base?
Conversation
Job PR-900-1 is done. |
PR looks awesome. Minor things about removing commented lines. Looking forward to the numbers trained on ImageNet. |
Job PR-900-2 is done. |
I'd like to hold this implementation from merged into our incoming release, unless one of the following can be met:
The reason behind is due to MXNet has a limited implementation of |
@hetong007 split may not support uneven division, but @becauseofAI used workaround which uses |
OK you are right, |
@becauseofAI any update on this? |
1 similar comment
@becauseofAI any update on this? |
|
||
def _split_channels(total_filters, num_groups): | ||
"""Get groups list.""" | ||
split_channels = [total_filters // num_groups for _ in range(num_groups)] |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
[total_filters // num_groups] * num_groups
Much faster
Add a new model named MixNet to gluoncv model-zoo.
It is a reimplementation of MixNet in Gluon. The paper is here. The model will be trained with ImageNet dataset when the machine is ready.