def __init__(self, model, freq=1): '''Initialize the MVCallback class The `model` should be the be a keras model The `freq` should be the update frequency of the parameters. For example, `freq=3` means update the parameters every 3 mini-batch. ''' super(MVCallback, self).__init__() self.kpm = KerasParamManager(model) self.cur_n = 0 if freq < 0: raise ValueError("Frequency must be an integer greater than 0.") self.freq = freq
def __init__(self, model): super(MVCallback, self).__init__() self.kpm = KerasParamManager(model)