f18c55e373aeddcecd3f1af90347d9a418d964f5,sonnet/src/optimizers/adam_test.py,,,#,138

Before Change

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


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

After Change

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

CONFIGS = optimizer_tests.named_product(learning_rate=(0.1, 0.01, 0.001),
                                        beta_1=(0.9, 0.99, 0.999),
                                        beta_2=(0.9, 0.99, 0.999),
                                        epsilon=(1e-8,),
                                        seed=(28, 2, 90))


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

Instances