Update loss.py

rvandeghen · web-flow · commit 71a7975a063e · 2020-04-30T17:04:15.000+02:00
correct typo
diff --git a/efficientdet/loss.py b/efficientdet/loss.py
@@ -57,9 +57,8 @@ def forward(self, classifications, regressions, anchors, annotations, **kwargs):
                 if torch.cuda.is_available()
                     
                     alpha_factor = torch.ones_like(classification) * alpha
-                    targets = targets.cuda()
                     alpha_factor = alpha_factor.cuda()
-                    alphe_factot = 1. - alpha_factor
+                    alpha_factor = 1. - alpha_factor
                     focal_weight = classification
                     focal_weight = alpha_factor * torch.pow(focal_weight, gamma)
                     
@@ -72,7 +71,7 @@ def forward(self, classifications, regressions, anchors, annotations, **kwargs):
                 else:
                     
                     alpha_factor = torch.ones_like(classification) * alpha
-                    alphe_factot = 1. - alpha_factor
+                    alpha_factor = 1. - alpha_factor
                     focal_weight = classification
                     focal_weight = alpha_factor * torch.pow(focal_weight, gamma)
                     
