You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
...
The text was updated successfully, but these errors were encountered:
Description
I think in
resnet.py
, the padding option forMaxPool
should be'SAME'
. The shape of the output ofMaxPool
and theResnet50ConvBlock
right after it becomesB x 55 x 55 x C
instead ofB x 56 x 56 x C
. See Keras and PyTorch....
Environment information
For bugs: reproduction and error logs
The text was updated successfully, but these errors were encountered: