force R=1 when object count is known

Former-commit-id: e5c7498f5be4b747f61daef3481b2606bcdb60d6
javiribera · Nov 5, 2018 · 3fd4ca4 · 3fd4ca4
1 parent dcb57d3
commit 3fd4ca4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/object-locator/logger.py b/object-locator/logger.py
@@ -137,7 +137,8 @@ def val_losses(self, terms, iteration_number, terms_legends=None):
         curated_terms = []
         for term in terms:
             if isinstance(term, numbers.Number):
-                curated_term = torch.tensor([term])
+                curated_term = torch.tensor([term],
+                                            dtype=torch.get_default_dtype())
             elif isinstance(term, torch.Tensor):
                 curated_term = term
             else:

diff --git a/object-locator/train.py b/object-locator/train.py
@@ -439,7 +439,8 @@
                           judge.rmse,
                           judge.mape,
                           judge.coeff_of_determination,
-                          judge.pearson_corr,
+                          judge.pearson_corr \
+                              if not np.isnan(judge.pearson_corr) else 1,
                           judge.precision,
                           judge.recall),
                    iteration_number=epoch,