(self, num_classes=12, input_size=2048, temp=0.05, norm=True)
| 81 | |
| 82 | class ResClassifier_MME(nn.Module): |
| 83 | def __init__(self, num_classes=12, input_size=2048, temp=0.05, norm=True): |
| 84 | super(ResClassifier_MME, self).__init__() |
| 85 | if norm: |
| 86 | self.fc = nn.Linear(input_size, num_classes, bias=False) |
| 87 | else: |
| 88 | self.fc = nn.Linear(input_size, num_classes, bias=False) |
| 89 | self.norm = norm |
| 90 | self.tmp = temp |
| 91 | |
| 92 | def set_lambda(self, lambd): |
| 93 | self.lambd = lambd |