From 233adeb7ae2f5a59ebbe0b3616a67eab75a30e44 Mon Sep 17 00:00:00 2001 From: Silviu Marian Udrescu Date: Sun, 26 Apr 2020 02:10:36 -0400 Subject: [PATCH] Add files via upload --- Code/S_NN_train.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/Code/S_NN_train.py b/Code/S_NN_train.py index 48d646b..652b17f 100644 --- a/Code/S_NN_train.py +++ b/Code/S_NN_train.py @@ -137,13 +137,17 @@ def NN_train(pathdir, filename, epochs=1000, lrs=1e-2, N_red_lr=4, pretrained_pa optimizer_feynman.step() # Early stopping - if epoch%20==0: + if epoch%20==0 and epoch>0: if check_es_loss < loss: break else: torch.save(model_feynman.state_dict(), "results/NN_trained_models/models/" + filename + ".h5") check_es_loss = loss - + if epoch==0: + if check_es_loss < loss: + torch.save(model_feynman.state_dict(), "results/NN_trained_models/models/" + filename + ".h5") + check_es_loss = loss + print(loss) lrs = lrs/10