remove until fixed

Parry-Parry · Dec 14, 2024 · 9f0fc70 · 9f0fc70
1 parent 0cdf522
commit 9f0fc70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rankers/train/flax/trainer.py b/rankers/train/flax/trainer.py
@@ -389,7 +389,7 @@ def get_learning_rates(self):
                 "Trainer optimizer is None, please make sure you have setup the optimizer before."
             )
         return [group["lr"] for group in self.optimizer.param_groups]
-
+    '''
     def _load_best_model(self):
         logger.info(f"Loading best model from {self.state.best_model_checkpoint} (score: {self.state.best_metric}).")
         best_model_path = os.path.join(self.state.best_model_checkpoint, WEIGHTS_NAME)
@@ -469,7 +469,7 @@ def _load_best_model(self):
             load_result = load_sharded_checkpoint(
                 model, self.state.best_model_checkpoint
             )
-
+    '''
     def train(
         self,
         resume_from_checkpoint: Optional[Union[str, bool]] = None,