google / flax

Flax is a neural network library for JAX that is designed for flexibility.

Home Page:https://flax.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Document default stride for pooling functions

marcvanzee opened this issue · comments

Discussed in #2023

Originally posted by dogeplusplus April 3, 2022
A bit of a nitpick but I was wondering why the default behavior of pooling functions is to have stride 1 instead of the window_shape? I feel that for most use cases the stride would be the dimension of the kernel size as in other frameworks.