Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
MsterDC authored Jan 19, 2022
1 parent 2878cf8 commit 7bc96d8
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions models/google/inceptionv3_sst.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,6 @@


def model(pretrained=False, **kwargs):
r"""Inception v3 model architecture from
`"Rethinking the Inception Architecture for Computer Vision" <http://arxiv.org/abs/1512.00567>`_.
Args:
pretrained (bool): If True, returns a model pre-trained on ImageNet
"""
if pretrained:
model = Inception3(**kwargs)
model_dict = model.state_dict()
Expand Down Expand Up @@ -271,7 +265,6 @@ def get_sos_loss(self, pre_hm, gt_hm):
def get_cls_loss(self, logits, label):
return self.ce_loss(logits, label.long())


def get_ra_loss(self, logits, label, th_bg=0.3, bg_fg_gap=0.0):
n, _, _, _ = logits.size()
cls_logits = F.softmax(logits, dim=1)
Expand Down

0 comments on commit 7bc96d8

Please sign in to comment.