A reimplementation of MixNet in Gluon.#900
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.
[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.