[losses] Remove useless '_' in public function.

Author: arthurdouillard

inclearn/lib/utils.py

@@ -17,7 +17,7 @@ def to_onehot(targets, n_classes):
     return onehot
 
 
-def _check_loss(loss):
+def check_loss(loss):
     return not bool(torch.isnan(loss).item()) and bool((loss >= 0.).item())
 
 
@@ -168,8 +168,9 @@ def apply_kmeans(features, targets, nb_clusters, pre_normalization):
     return np.concatenate(new_features), np.concatenate(new_targets)
 
 
-def apply_knn(features, targets, features_test, targets_test, nb_neighbors, normalize=True,
-              weights="uniform"):
+def apply_knn(
+    features, targets, features_test, targets_test, nb_neighbors, normalize=True, weights="uniform"
+):
     print(
         "KNN with {} neighbors and pre-normalized features: {}, weights: {}.".format(
             nb_neighbors, normalize, weights

inclearn/models/icarl.py

@@ -237,7 +237,7 @@ def _forward_loss(self, training_network, inputs, targets, memory_flags):
 
         loss = self._compute_loss(inputs, outputs, targets, onehot_targets, memory_flags)
 
-        if not utils._check_loss(loss):
+        if not utils.check_loss(loss):
             pdb.set_trace()
 
         self._metrics["loss"] += loss.item()