padding has incorrect dimension for `nn.AvgPool2d`'s `def __init__()`

[dyollb]

model-zoo/models/wholeBrainSeg_Large_UNEST_segmentation/scripts/networks/nest/utils.py

Line 361 in d86a119

super(AvgPool3dSame, self).__init__(kernel_size, stride, (0, 0, 0), ceil_mode, count_include_pad)

Running mypy yields arg-type error.

See https://docs.pytorch.org/docs/stable/generated/torch.nn.AvgPool2d.html#avgpool2d

[ericspod]

@tangy5 I think this bundle was contributed by you, could you please have a look at this? Thanks!