d35d95925d72f4bf83d4eaeb66ec3e93a1b3a663,src/garage/tf/optimizers/first_order_optimizer.py,FirstOrderOptimizer,__init__,#FirstOrderOptimizer#Any#Any#Any#Any#Any#Any#Any#Any#,32

Before Change

        if tf_optimizer_cls is None:
            tf_optimizer_cls = tf.compat.v1.train.AdamOptimizer
        if tf_optimizer_args is None:
            tf_optimizer_args = dict(learning_rate=1e-3)
        self._tf_optimizer = tf_optimizer_cls(**tf_optimizer_args)
        self._max_epochs = max_epochs
        self._tolerance = tolerance
        self._batch_size = batch_size

After Change

        if optimizer is None:
            optimizer = tf.compat.v1.train.AdamOptimizer
        learning_rate = learning_rate or dict(learning_rate=_Default(1e-3))
        if not isinstance(learning_rate, dict):
            learning_rate = dict(learning_rate=learning_rate)

        self._tf_optimizer = optimizer
        self._learning_rate = learning_rate
        self._max_epochs = max_epochs
        self._tolerance = tolerance

Instances