Skip to content

Commit afd229e

Browse files
committed
[tmva][sofie] Use -1 as default for batch_size in RFunction_Update::GenerateModel
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
1 parent 1b77790 commit afd229e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tmva/sofie/inc/TMVA/RFunction.hxx

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,7 @@ public:
5252
std::shared_ptr<RModel> GetFunctionBlock() {
5353
return function_block;
5454
}
55-
56-
std::string GenerateModel(const std::string& filename, long read_pos=0, long block_size=1);
55+
std::string GenerateModel(const std::string& filename, long read_pos = 0, long block_size = -1);
5756
std::string Generate(const std::vector<std::string>& inputPtrs);
5857
FunctionTarget GetFunctionTarget() {
5958
return fTarget;

0 commit comments

Comments
 (0)