diff --git a/hourglass_tensorflow/handlers/train.py b/hourglass_tensorflow/handlers/train.py
index 2255169..096470f 100644
--- a/hourglass_tensorflow/handlers/train.py
+++ b/hourglass_tensorflow/handlers/train.py
@@ -38,6 +38,7 @@ def __init__(
         self._optimizer: Union[HTFObjectReference[Optimizer], str] = None
         self._metrics: List[HTFObjectReference[Metric]] = None
         self._callbacks: List[HTFObjectReference[Callback]] = None
+        self.init_handler()
 
     @property
     def config(self) -> HTFTrainConfig: