indicate when len(mask)==0

Former-commit-id: 9ab2ab9ea9ee3d42e1d571d01051ce87821b9775
javiribera · Mar 7, 2018 · d6a9b15 · d6a9b15
1 parent d5264bc
commit d6a9b15
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plant-locator/train.py b/plant-locator/train.py
@@ -330,6 +330,7 @@
         if len(c) == 0:
             ahd = criterion_training.max_dist
             centroids = []
+            print('len(c) == 0')
         else:
             n_components = int(torch.round(est_count).data.cpu().numpy()[0])
             # If the estimation is horrible, we cannot fit a GMM if n_components > n_samples