| 86 | |
| 87 | |
| 88 | class VGG16(BaseNet): |
| 89 | def __init__(self): |
| 90 | super(VGG16, self).__init__() |
| 91 | |
| 92 | self.layers = models.vgg16(weights=models.VGG16_Weights.IMAGENET1K_V1).features |
| 93 | self.target_layers = [4, 9, 16, 23, 30] |
| 94 | self.n_channels_list = [64, 128, 256, 512, 512] |
| 95 | |
| 96 | self.set_requires_grad(False) |