From 9b5cbcfee4fd65747f74c3ed38be3c7181382f5a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?D=C5=BEenan=20Zuki=C4=87?= Date: Sun, 17 Apr 2022 11:30:52 -0400 Subject: [PATCH] Reduce number of epochs slightly --- miqa/learning/nn_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/miqa/learning/nn_training.py b/miqa/learning/nn_training.py index f504d523..bc081c13 100755 --- a/miqa/learning/nn_training.py +++ b/miqa/learning/nn_training.py @@ -644,7 +644,7 @@ def process_folds(folds_prefix, validation_fold, evaluate_only, fold_count): # establish minimum number of optimization steps and epochs val_count = max(1, int(600 / df.shape[0])) - epoch_count = max(40, int(30000 / df.shape[0])) + epoch_count = max(35, int(30000 / df.shape[0])) epoch_count = math.ceil(epoch_count / val_count) * val_count count_train = df.shape[0] - vf.shape[0]