add get early stopping method

Author: niklasrah

yolox/core/trainer.py

@@ -76,6 +76,7 @@ def __init__(self, exp: Exp, args):
         # before_train methods.
         self.exp = exp
         self.args = args
+        self.early_stopper = None
 
         # training related attr
         self.max_epoch = exp.max_epoch

yolox/exp/yolox_base.py

@@ -349,6 +349,11 @@ def get_trainer(self, args):
         # NOTE: trainer shouldn't be an attribute of exp object
         return trainer
 
+    def get_early_stopping(self, patience, min_delta, mode):
+        from yolox.core import EarlyStopping
+
+        return EarlyStopping(patience=patience, min_delta=min_delta, mode=mode)
+
     def eval(self, model, evaluator, is_distributed, half=False, return_outputs=False):
         return evaluator.evaluate(model, is_distributed, half, return_outputs=return_outputs)