-
-
Notifications
You must be signed in to change notification settings - Fork 608
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
Create additional pooling layers #1224
Comments
Great! A couple of notes:
|
Oh, I should remove AvgPool then. |
And probably want to name the layer AdaptiveMeanPool for consistency. |
shall i make required changes in the NNlib.jl and create adaptivemaxpool and adaptivemeanpool backends too? I will add two functions there only. |
I don’t think that’s necessary. Adaptive pooling does the same operation as regular pooling except that the filter window is adjusted dynamically to account for input image size. The pooling functions in NNlib.jl accept a |
1239: add adaptive pool r=CarloLucibello a=dnabanita7 I have added ``AdaptiveMaxPool`` and ``AdaptiveMeanPool`` so that we can do a similar [PyTorch implementation](darsnack/FluxModels.jl#1 (comment)). cc @darsnack ### PR - [x] Tests are added - [x] Entry in NEWS.md - [x] Documentation, if applicable - [ ] Final review from `@MikeInnes` or `@dhairyagandhi96` (for API changes). ### Flux issue linking [Flux#1224](#1224) ### MLH issue linking [0.3.x-projects#26](https://github.com/MLH-Fellowship/0.3.x-projects/issues/26) Co-authored-by: Nabanita Dash <dashnabanita@gmail.com>
Fixed by #1239 |
This issue is in for better co-ordination of the FluxModels.jl which can be used as a benchmark for Flux.jl. As explained by @darsnack in this issue, I will be creating the following pooling layers.
The text was updated successfully, but these errors were encountered: