From 3eda9edb4931ff5c0dd05369ef3ee375fd5cc38b Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 18 Aug 2023 09:58:27 +0000
Subject: [PATCH] Remove obsolete curriculum
---
dan/ocr/manager/training.py | 10 ----------
1 file changed, 10 deletions(-)
diff --git a/dan/ocr/manager/training.py b/dan/ocr/manager/training.py
index e1c9871d..c50ff33a 100644
--- a/dan/ocr/manager/training.py
+++ b/dan/ocr/manager/training.py
@@ -689,16 +689,6 @@ class GenericTrainingManager:
self.save_model(epoch=num_epoch, name="best")
self.best = eval_values["cer"]
- # Handle curriculum learning update
- if self.dataset.train_dataset.curriculum_config:
- self.check_and_update_curriculum()
-
- if (
- "curriculum_model" in self.params["model"]
- and self.params["model"]["curriculum_model"]
- ):
- self.update_curriculum_model()
-
# save model weights
if self.is_master:
self.save_model(epoch=num_epoch, name="last")
--
GitLab