Commit 5854b9ca authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix adagrad

parent e525f10e
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -21,7 +21,7 @@ class Optimizer(object):
    -------
    a new TensorFlow optimizer implementing the algorithm
    """
    raise NotImplemented("Subclasses must implement this")
    raise NotImplementedError("Subclasses must implement this")


class LearningRateSchedule(object):
@@ -42,7 +42,7 @@ class LearningRateSchedule(object):
    -------
    a tensor that equals the learning rate
    """
    raise NotImplemented("Subclasses must implement this")
    raise NotImplementedError("Subclasses must implement this")


class AdaGrad(Optimizer):
@@ -85,7 +85,7 @@ learning research 12.7 (2011).
    else:
      learning_rate = self.learning_rate
    return tf.keras.optimizers.Adagrad(
        learning_rate=self.learning_rate,
        learning_rate=learning_rate,
        initial_accumulator_value=self.initial_accumulator_value,
        epsilon=self.epsilon)