-
Notifications
You must be signed in to change notification settings - Fork 1k
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
Solves #1512 #1526
Solves #1512 #1526
Conversation
@awni We have implemented Maxpool3d and Avgpool3d, if it looks good then we can proceed with adding tests to it. |
# Conflicts: # python/mlx/nn/layers/pooling.py
I think after merging the pooling was removed? Could you add pool3d back, rebase on main and push force to the same branch? Then I will run the tests and merge if they pass. Also, sorry for the nitpicking but do you mind putting Thanks! |
# Conflicts: # python/mlx/nn/layers/pooling.py
@angeloskath We have made the suggested changes. I think this is ready to merge. Let me know if anything else is required from our side. |
@angeloskath do you think anything needs any correction? |
@angeloskath @awni Could you confirm if everything is fine, and if so, would it be possible to merge it? |
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.
Looks good!
Proposed changes
Solves #1512
Added pool3d class and maxpool3d in pooling.py file
Please include a description of the problem or feature this PR is addressing. If there is a corresponding issue, include the issue #.
Checklist
Put an
x
in the boxes that apply.pre-commit run --all-files
to format my code / installed pre-commit prior to committing changes