Oneway to do it would be pull the source code from pytorch and change the layers as you want.
Eg: https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py 265
pull the resent and change resnet as you need