kazuto1011 / deeplab-pytorch

PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Why do you set ceil_mode=True in the _Stem class of the file resnet.py?

cnyanhao opened this issue · comments

Thank you for your legible and powerful code! I've noticed that you set ceil_mode=True in the _Stem class of the file resnet.py, here. This will change the size of the output feature. I've noticed that the implementation of torchvision didn't use that, here. Do you have any special consideration to add this? Thank you so much.

My ResNet modules are defined to match the original DeepLab v2 written in Caffe to load a pre-trained caffemodel. The "ceil" pooling is from here in the authors' Caffe. The bottleneck block is also different from the one in torchvision, on whether the strided-downsampling is in the 1st 1x1 or 2nd 3x3 convolution.