| 75 | |
| 76 | |
| 77 | class AlexNet(BaseNet): |
| 78 | def __init__(self): |
| 79 | super(AlexNet, self).__init__() |
| 80 | |
| 81 | self.layers = models.alexnet(True).features |
| 82 | self.target_layers = [2, 5, 8, 10, 12] |
| 83 | self.n_channels_list = [64, 192, 384, 256, 256] |
| 84 | |
| 85 | self.set_requires_grad(False) |
| 86 | |
| 87 | |
| 88 | class VGG16(BaseNet): |