diff options
Diffstat (limited to 'tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py')
-rw-r--r-- | tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py b/tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py index 0824ecf616..e89993991a 100644 --- a/tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py +++ b/tensorflow/contrib/gan/python/estimator/python/gan_estimator_impl.py @@ -76,7 +76,7 @@ class GANEstimator(estimator.Estimator): return logits # Create GAN estimator. - gan_estimator = tfgan.estimator.GANEstimator( + gan_estimator = estimator.GANEstimator( model_dir, generator_fn=generator_fn, discriminator_fn=discriminator_fn, |