Esempio n. 1
0
    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
Esempio n. 2
0
 def __init__(self, model):
     super(MVCallback, self).__init__()
     self.kpm = KerasParamManager(model)