resnet_strides: add strides as a parameter to the ResNet like in the PyTorch. Sometimes it's necessary to change it. For example in this paper https://arxiv.org/abs/2103.17263 on the evaluation state strides have to be (1, 2, 1, 1).
resnet_strides: add strides as a parameter to the ResNet like in the PyTorch. Sometimes it's necessary to change it. For example in this paper https://arxiv.org/abs/2103.17263 on the evaluation state strides have to be (1, 2, 1, 1).

deepmind/dm-haiku,deepmind/dm-haiku