We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
https://github.com/ZiyueHuang/MXShuffleNet/blob/master/symbol.py#L11
In shuffle operation, the code should be
data = mx.sym.reshape(data, shape=(0, -4, channels // groups, -1, -2))
For example, (256, 60, 28, 28) should be reshaped to (256, 20, 3, 28, 28) and then we swap axes to (256, 3, 20, 28, 28) and reshape back.
The text was updated successfully, but these errors were encountered:
No branches or pull requests
https://github.com/ZiyueHuang/MXShuffleNet/blob/master/symbol.py#L11
In shuffle operation, the code should be
data = mx.sym.reshape(data, shape=(0, -4, channels // groups, -1, -2))
For example,
(256, 60, 28, 28) should be reshaped to (256, 20, 3, 28, 28) and then we swap axes to (256, 3, 20, 28, 28) and reshape back.
The text was updated successfully, but these errors were encountered: