[tmva][sofie] Apply fixes when batch size is given by user

When input shape is parametric, give possiibility to fix batch size if provided by user in RModel::Generate

Author: lmoneta

tmva/sofie/src/RModel.cxx

@@ -262,7 +262,8 @@ void RModel::Initialize(int batchSize, bool verbose) {
    // loop on inputs and see if shape can be  full specified
    // if the batch size is provided it can be used to specify the full shape
    // Add the full specified tensors in fReadyInputTensors collection
-   for (auto &input : fInputTensorInfos) {
+   auto originalInputTensorInfos = fInputTensorInfos; // need to copy because we may delete elements
+   for (auto &input : originalInputTensorInfos) {
       std::cout << "looking at the tensor " << input.first << std::endl;
       // if a batch size is provided convert batch size
       // assume is parametrized as "bs" or "batch_size"
@@ -288,12 +289,10 @@ void RModel::Initialize(int batchSize, bool verbose) {
       }
       auto shape = ConvertShapeToInt(input.second.shape);
       if (!shape.empty()) {
-#if 0
          // remove from the tensor info old dynamic shape
          fInputTensorInfos.erase(input.first);
          // add to the ready input tensor information the new fixed shape
-         AddInputTensorInfo(input.first, input.second.type, shape);  
-#endif
+         AddInputTensorInfo(input.first, input.second.type, shape);
       }
       // store the parameters of the input tensors
       else {
@@ -647,7 +646,7 @@ void RModel::ReadInitializedTensorsFromFile(long pos) {
         fGC += "   std::ifstream f;\n";
         fGC += "   f.open(filename);\n";
         fGC += "   if (!f.is_open()) {\n";
-        fGC += "      throw std::runtime_error(\"tmva-sofie failed to open file for input weights\");\n";
+        fGC += "      throw std::runtime_error(\"tmva-sofie failed to open file \" + filename + \" for input weights\");\n";
         fGC += "   }\n";
 
         if(fIsGNNComponent) {
@@ -783,7 +782,7 @@ long RModel::WriteInitializedTensorsToFile(std::string filename) {
         }
         if (!f.is_open())
             throw
-            std::runtime_error("tmva-sofie failed to open file for tensor weight data");
+            std::runtime_error("tmva-sofie failed to open file " + filename + " for tensor weight data");
         for (auto& i: fInitializedTensors) {
             if (i.second.type() == ETensorType::FLOAT) {
                 size_t length = 1;