fix(torch): metrics naming for multiple test sets

Author: Bycob

src/backends/torch/torchlib.cc

@@ -1209,8 +1209,11 @@ namespace dd
                     std::vector<std::string> meas_names = meas_obj.list_keys();
                     for (auto name : meas_names)
                       {
-                        std::string metric_name
-                            = name + "_test" + std::to_string(i);
+                        std::string metric_name;
+                        if (i == 0)
+                          metric_name = name;
+                        else
+                          metric_name = name + "_test" + std::to_string(i - 1);
 
                         if (name != "cmdiag" && name != "cmfull"
                             && name != "clacc" && name != "cliou"

tests/ut-torchapi.cc

@@ -929,6 +929,14 @@ TEST(torchapi, service_train_resume)
     }
 
   // remove files
+  ASSERT_TRUE(fileops::file_exists(resnet50_train_repo + "best_model.txt"));
+  remove((resnet50_train_repo + "best_model.txt").c_str());
+  ASSERT_TRUE(
+      fileops::file_exists(resnet50_train_repo + "best_model_test_0.txt"));
+  remove((resnet50_train_repo + "best_model_test_0.txt").c_str());
+  ASSERT_FALSE(
+      fileops::file_exists(resnet50_train_repo + "best_model_test_1.txt"));
+
   std::unordered_set<std::string> lfiles;
   fileops::list_directory(resnet50_train_repo, true, false, false, lfiles);
   for (std::string ff : lfiles)