From 890ab3a412725220628089ea1fc7e50b466bc485 Mon Sep 17 00:00:00 2001
From: moneta <lorenzo.moneta@cern.ch>
Date: Tue, 17 Jan 2023 10:11:47 +0100
Subject: [PATCH] [tmva] More fixes to avoid timeout in TMVA tutorials
---
tutorials/CMakeLists.txt | 2 +-
tutorials/tmva/TMVAMulticlass.C | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/tutorials/CMakeLists.txt b/tutorials/CMakeLists.txt
index bdda28f179e..07f53f2e356 100644
--- a/tutorials/CMakeLists.txt
+++ b/tutorials/CMakeLists.txt
@@ -569,7 +569,7 @@ file(GLOB long_running RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} ${long_running})
set (multithreaded
dataframe/df10[2-7]*
multicore/mp103*
- tmva/TMVAMultiClass.C
+ tmva/TMVAMulticlass.C
tmva/TMVA_CNN_Classification.C
tmva/TMVA_Higgs_Classification.C
tmva/TMVA_RNN_Classification.C
diff --git a/tutorials/tmva/TMVAMulticlass.C b/tutorials/tmva/TMVAMulticlass.C
index 09c70ac2be6..7194af54e35 100644
--- a/tutorials/tmva/TMVAMulticlass.C
+++ b/tutorials/tmva/TMVAMulticlass.C
@@ -137,7 +137,7 @@ void TMVAMulticlass( TString myMethodList = "" )
if (Use["DL_CPU"]) {
TString layoutString("Layout=TANH|100,TANH|50,TANH|10,LINEAR");
TString trainingStrategyString("TrainingStrategy=Optimizer=ADAM,LearningRate=1e-3,"
- "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100");
+ "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100,MaxEpochs=20");
TString nnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=N:"
"WeightInitialization=XAVIERUNIFORM:Architecture=GPU");
nnOptions.Append(":");
@@ -149,7 +149,7 @@ void TMVAMulticlass( TString myMethodList = "" )
if (Use["DL_GPU"]) {
TString layoutString("Layout=TANH|100,TANH|50,TANH|10,LINEAR");
TString trainingStrategyString("TrainingStrategy=Optimizer=ADAM,LearningRate=1e-3,"
- "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100");
+ "TestRepetitions=1,ConvergenceSteps=10,BatchSize=100,MaxEpochs=20");
TString nnOptions("!H:V:ErrorStrategy=CROSSENTROPY:VarTransform=N:"
"WeightInitialization=XAVIERUNIFORM:Architecture=GPU");
nnOptions.Append(":");
--
GitLab