Possible padding bug in Resnet50

[ryotatomioka]

Description

I think in resnet.py, the padding option for MaxPool should be 'SAME'. The shape of the output of MaxPool and the Resnet50ConvBlock right after it becomes B x 55 x 55 x C instead of B x 56 x 56 x C. See Keras and PyTorch.

...

Environment information

OS: <your answer here>

$ pip freeze | grep tensor
# your output here

$ pip freeze | grep jax
# your output here

$ python -V
# your output here

For bugs: reproduction and error logs

# Steps to reproduce:
...

# Error logs:
...