I’m trying to implement PSPNet (Pyramid Scene Parsing Network). The network uses resnet with dilation and it is trained on imagenet. Is there way to directly use pre-trained torchvision ResNet layers [and inject the dilation into required Conv2D modules] rather than building the layers and then transferring the weights ?