mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add softmax to spinenet classification head

Browse Source
This commit is contained in:
James Betker 2020-12-16 09:42:15 -07:00
parent 4310e66848
commit 09de3052ac
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/archs/spinenet_arch.py
View File

@ -353,7 +353,8 @@ class SpinenetWithLogits(SpineNet):
self.output_to_attach = output_to_attach self.output_to_attach = output_to_attach
self.tail = nn.Sequential(ConvBnRelu(256, 128, kernel_size=1, activation=True, norm=True, bias=False), self.tail = nn.Sequential(ConvBnRelu(256, 128, kernel_size=1, activation=True, norm=True, bias=False),
ConvBnRelu(128, 64, kernel_size=1, activation=True, norm=True, bias=False), ConvBnRelu(128, 64, kernel_size=1, activation=True, norm=True, bias=False),
ConvBnRelu(64, num_labels, kernel_size=1, activation=False, norm=False, bias=True)) ConvBnRelu(64, num_labels, kernel_size=1, activation=False, norm=False, bias=True),
nn.Softmax(dim=1))
def forward(self, x): def forward(self, x):
fea = super().forward(x)[self.output_to_attach] fea = super().forward(x)[self.output_to_attach]