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")