You could copy the VGG
implementation from torchvision.models.vgg
and change the layers as you want, which would be the simplest approach for your custom architecture.
You could copy the VGG
implementation from torchvision.models.vgg
and change the layers as you want, which would be the simplest approach for your custom architecture.