From 2dc49f5ccb4db4f002f446c9efce82ad6b89177e Mon Sep 17 00:00:00 2001 From: Tim de Bruin Date: Fri, 10 Jul 2020 10:28:45 +0100 Subject: [PATCH] initial stage fix (#195) --- .../training/knowledge_distillation/multi_stage_training.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/larq_zoo/training/knowledge_distillation/multi_stage_training.py b/larq_zoo/training/knowledge_distillation/multi_stage_training.py index e751a226..e5dfa03e 100644 --- a/larq_zoo/training/knowledge_distillation/multi_stage_training.py +++ b/larq_zoo/training/knowledge_distillation/multi_stage_training.py @@ -179,5 +179,7 @@ def run(self) -> None: Path(self.parent_output_dir).mkdir(parents=True, exist_ok=True) for experiment in self.experiments: + if experiment.stage < self.initial_stage: + continue print(f"Starting stage {experiment.stage} at {datetime.now().isoformat()}.") experiment.run()