fix: torch best model better or equal

jolibrain · Mar 22, 2022 · 4d50c8e · 4d50c8e
1 parent 2dabd89
commit 4d50c8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -481,8 +481,8 @@ namespace dd
   {
     if (metric_name == "eucll" || metric_name == "delta_score_0.1"
         || metric_name == "L1_mean_error")
-      return (v2 > v1);
-    return (v1 > v2);
+      return (v2 >= v1);
+    return (v1 >= v2);
   }
 
   template <class TInputConnectorStrategy, class TOutputConnectorStrategy,