From 514a897dd5392bf7a160bf02c82b04da1fc53bb0 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Sun, 8 Feb 2026 01:11:57 +0100 Subject: [PATCH] training.train: assert n_epochs vs. index_start --- src/eynollah/training/train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/eynollah/training/train.py b/src/eynollah/training/train.py index ecf70b4..73d5e0b 100644 --- a/src/eynollah/training/train.py +++ b/src/eynollah/training/train.py @@ -275,6 +275,9 @@ def run(_config, run configured experiment via sacred """ + if continue_training: + assert n_epochs > index_start, "with continue_training, n_epochs must be greater than index_start" + if pretraining and not os.path.isfile(RESNET50_WEIGHTS_PATH): _log.info("downloading RESNET50 pretrained weights to %s", RESNET50_WEIGHTS_PATH) download_file(RESNET50_WEIGHTS_URL, RESNET50_WEIGHTS_PATH)