Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Shank2358 committed Jul 22, 2021
1 parent 4eabad8 commit 52d2742
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
Binary file added log/event/events.out.tfevents.1626962460.Shank
Binary file not shown.
3 changes: 1 addition & 2 deletions modelR/head/dsc_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,8 @@ def forward(self, input):
return (torch.clamp(torch.sigmoid(input),0.011,1)-0.01)/(1-0.01)

class DSC_Head(nn.Module):
def __init__(self, nC, anchors, stride, fact):
def __init__(self, nC, anchors, stride):
super(DSC_Head, self).__init__()
self.fact = fact
self.__anchors = anchors
self.__nA = len(anchors)
self.__nC = nC
Expand Down

0 comments on commit 52d2742

Please sign in to comment.