def METHOD_NAME(self, **kwargs: kwargus) -> None:
    self.permit_train = True
    for i in range(self.count):
        if not self.model_trained[i]:
            dict_kwargs = kwargs.copy()
            # TODO(@refraction-ray): still not compatible with new optimizer
            self.models[i].METHOD_NAME(**dict_kwargs)