Update main.py

filippo-masi · Jan 30, 2021 · d9183f1 · d9183f1
1 parent 5a08631
commit d9183f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")