Upsampling/Unpooling for Transpose Convolutions

patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Apache License 2.0

2.12k stars 142 forks source link

Upsampling/Unpooling for Transpose Convolutions #894

Open JadM133 opened 1 week ago

JadM133 commented 1 week ago

Hello all, Great work, the library is amazing. I am having trouble in finding unpooling classes. Did I miss them or are they unavailable yet? Thanks in advance!

patrick-kidger commented 1 week ago

I think you're the first one to ask for them! :)

I'd be happy to take a pull request that adds these.