Skip to content
Snippets Groups Projects
Commit 890ab3a4 authored by Lorenzo Moneta's avatar Lorenzo Moneta
Browse files

[tmva] More fixes to avoid timeout in TMVA tutorials

parent 4c9c5d6b
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
......@@ -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(":");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment