initial stage fix (#195)

larq · Jul 10, 2020 · 2dc49f5 · 2dc49f5
1 parent fd1c9d4
commit 2dc49f5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()