qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-02-20 16:32:03 +01:00
Code Issues Projects Releases Wiki Activity

training.train: assert n_epochs vs. index_start

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-08 01:11:57 +01:00
parent 37338049af
commit 514a897dd5
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/training/train.py
View file

@ -275,6 +275,9 @@ def run(_config,
run configured experiment via sacred 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): if pretraining and not os.path.isfile(RESNET50_WEIGHTS_PATH):
_log.info("downloading RESNET50 pretrained weights to %s", RESNET50_WEIGHTS_PATH) _log.info("downloading RESNET50 pretrained weights to %s", RESNET50_WEIGHTS_PATH)
download_file(RESNET50_WEIGHTS_URL, RESNET50_WEIGHTS_PATH) download_file(RESNET50_WEIGHTS_URL, RESNET50_WEIGHTS_PATH)