是否有一种方法可以使用预训练权重将空间变换器神经网络包含在预训练模型中，而无需重写完整的体系结构？
例如，在pytorch中
我想在pytorch中向alexnet（alexnet = torchvision.models.alexnet（pretrained = True））添加一个空间转换器（如本教程https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html所示）。
是否有办法做到这一点，或者我必须重写整个体系结构并自己从头开始训练模型？
谢谢