f18c55e373aeddcecd3f1af90347d9a418d964f5,sonnet/src/optimizers/rmsprop_test.py,,,#,177

Before Change

    return optimizer_tests.WrappedTFOptimizer(tf.optimizers.RMSprop(**kwargs))


if __name__ == "__main__":
  // tf.enable_v2_behavior()
  tf.test.main()

After Change

from sonnet.src.optimizers import rmsprop
import tensorflow as tf

CONFIGS = optimizer_tests.named_product(learning_rate=(0.01, 0.001),
                                        decay=(0.8, 0.9),
                                        momentum=(0.0, 0.5),
                                        epsilon=(1e-7, 1e-8),
                                        centered=(False, True),
                                        seed=(28, 2, 90))


class ComparisonTest(optimizer_tests.AbstractFuzzTest):
  Ensures Sonnet optimizers have equivalent behavior to TensorFlow.

Instances