Hi,
Is there any way to add a spacial transformer network (seen in the tutorial: https://pytorch.org/tutorials/intermediate/spatial_transformer_tutorial.html) to a pre-trained model? or do we have to copy the architecture of the pretrained model in order to add it?
Thanks