fix code style

patrickeala · Jul 9, 2020 · 20ce288 · 20ce288
1 parent b877737
commit 20ce288
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lanenet_model/lanenet_back_end.py b/lanenet_model/lanenet_back_end.py
@@ -136,9 +136,9 @@ def compute_loss(self, binary_seg_logits, binary_label,
                     )
                 elif self._binary_loss_type == 'focal':
                     binary_segmenatation_loss = self._multi_category_focal_loss(
-                    onehot_labels=binary_label_onehot,
-                    logits=binary_seg_logits,
-                    classes_weights=inverse_weights
+                        onehot_labels=binary_label_onehot,
+                        logits=binary_seg_logits,
+                        classes_weights=inverse_weights
                     )
                 else:
                     raise NotImplementedError