fix(torch): add measures to output event when training not done

an empty metrics field breaks the platform, and metrics field remains empty when the training is stopped early
jolibrain · Mar 16, 2023 · 5714767 · 5714767
1 parent 4fe0df7
commit 5714767
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -1291,11 +1291,8 @@ namespace dd
                       }
                   }
 
-                if (elapsed_it == iterations)
-                  {
-                    out.add("measure", meas_out.getobj("measure"));
-                    out.add("measures", meas_out.getv("measures"));
-                  }
+                out.add("measure", meas_out.getobj("measure"));
+                out.add("measures", meas_out.getv("measures"));
               }
 
             train_loss = 0;