From d9183f1551507be30b2b297e607334abe262adc4 Mon Sep 17 00:00:00 2001 From: flpmasi <78264695+flpmasi@users.noreply.github.com> Date: Sat, 30 Jan 2021 15:25:28 +0100 Subject: [PATCH] Update main.py --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 2ca9d48..c5b2d56 100644 --- a/main.py +++ b/main.py @@ -18,7 +18,7 @@ # Training, evaluation against training, validation, and test data-sets, and weights export if silent==False: print("\n... Training") -learningRate=1e-4; nEpochs=1000; bSize=10 +learningRate=1e-4; nEpochs=100000; bSize=10 historyTraining=ThermoANN.setTraining(ThermoANN,uN_T,oN_T,uN_V,oN_V,learningRate,nEpochs,bSize) ThermoANN.evaluate(uN_T,oN_T); ThermoANN.evaluate(uN_V,oN_V); ThermoANN.evaluate(uN_Tt,oN_Tt) if silent==False: print("\n... Saving weights")