| 64 | |
| 65 | |
| 66 | class SqueezeNet(BaseNet): |
| 67 | def __init__(self): |
| 68 | super(SqueezeNet, self).__init__() |
| 69 | |
| 70 | self.layers = models.squeezenet1_1(True).features |
| 71 | self.target_layers = [2, 5, 8, 10, 11, 12, 13] |
| 72 | self.n_channels_list = [64, 128, 256, 384, 384, 512, 512] |
| 73 | |
| 74 | self.set_requires_grad(False) |
| 75 | |
| 76 | |
| 77 | class AlexNet(BaseNet): |