'YFOptimizer' object has no attribute '_h_min' when calling optimizer.state_dict()

When I try to save the optimizer.state_dict() before the first training step, this error occurred. Seems that we should add `self._h_min = 0.0` and `self._h_max = 0.0` in `__init__()` ?