Skip to content
Snippets Groups Projects
Unverified Commit 7ac6ebee authored by Yosuke Shinya's avatar Yosuke Shinya Committed by GitHub
Browse files

Update to support fp16 training of GFL (#3410)

parent cd37b1fb
No related branches found
No related tags found
No related merge requests found
......@@ -52,7 +52,7 @@ class Integral(nn.Module):
offsets from the box center in four directions, shape (N, 4).
"""
x = F.softmax(x.reshape(-1, self.reg_max + 1), dim=1)
x = F.linear(x, self.project).reshape(-1, 4)
x = F.linear(x, self.project.type_as(x)).reshape(-1, 4)
return x
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment