Flux.jl: Create additional pooling layers

Created on 13 Jun 2020  路  6Comments  路  Source: FluxML/Flux.jl

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.

  • [x] AdaptiveMeanPool
  • [x] AdaptiveMaxPool

All 6 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鈥檛 think that鈥檚 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 PoolDims object. Your layers in Flux should create this object based on the desired output size when layer(x) is called. Then pass that object to NNlib.

Fixed by #1239

Was this page helpful?
0 / 5 - 0 ratings

Related issues

sklan picture sklan  路  6Comments

jw3126 picture jw3126  路  6Comments

philip-bl picture philip-bl  路  3Comments

MikeInnes picture MikeInnes  路  3Comments

caseykneale picture caseykneale  路  5Comments