(self, num_classes=1000)
| 123 | |
| 124 | class FaceNet(nn.Module): |
| 125 | def __init__(self, num_classes=1000): |
| 126 | super(FaceNet, self).__init__() |
| 127 | self.feature = evolve.IR_50_112((112, 112)) |
| 128 | self.feat_dim = 512 |
| 129 | self.num_classes = num_classes |
| 130 | self.fc_layer = nn.Linear(self.feat_dim, self.num_classes) |
| 131 | |
| 132 | def predict(self, x): |
| 133 | feat = self.feature(x) |