Skip to content

Commit

Permalink
[tmva][sofie] Use -1 as default for batch_size in RFunction_Update::G…
Browse files Browse the repository at this point in the history
…enerateModel

The GenerateModel function called for the GNN classes was having a default =1 for the batch size used when generating the update models.
We should use -1 to not force any specific batch size and use directly the given input shape
  • Loading branch information
lmoneta committed Jun 13, 2024
1 parent 6feba92 commit 45d08d4
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tmva/sofie/inc/TMVA/RFunction.hxx
Original file line number Diff line number Diff line change
Expand Up @@ -52,8 +52,7 @@ public:
std::shared_ptr<RModel> GetFunctionBlock() {
return function_block;
}

std::string GenerateModel(const std::string& filename, long read_pos=0, long block_size=1);
std::string GenerateModel(const std::string& filename, long read_pos = 0, long block_size = -1);
std::string Generate(const std::vector<std::string>& inputPtrs);
FunctionTarget GetFunctionTarget() {
return fTarget;
Expand Down

0 comments on commit 45d08d4

Please sign in to comment.