schedualer.py

class BaseSchedualer():

    def __init__(self):
        self._prog = None
    
    def _set_prog(self, prog):
        self._prog = prog

    def build(self, learning_rate):
        raise NotImplementedError()