We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 74a68e3 commit ac3f68cCopy full SHA for ac3f68c
1 file changed
efficientdet/loss.py
@@ -54,7 +54,7 @@ def forward(self, classifications, regressions, anchors, annotations, **kwargs):
54
classification = torch.clamp(classification, 1e-4, 1.0 - 1e-4)
55
56
if bbox_annotation.shape[0] == 0:
57
- if torch.cuda.is_available()
+ if torch.cuda.is_available():
58
59
alpha_factor = torch.ones_like(classification) * alpha
60
alpha_factor = alpha_factor.cuda()
0 commit comments