Import inception_v3 from tf_slim.nets since it is not publicly visible.

Fixes #1751. PiperOrigin-RevId: 316736492 Change-Id: If915dc2ace5458dafc2909b1c49e086344227e29
magenta · Jun 16, 2020 · 36b050d · 36b050d
1 parent 050460f
commit 36b050d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/magenta/models/arbitrary_image_stylization/arbitrary_image_stylization_build_model.py b/magenta/models/arbitrary_image_stylization/arbitrary_image_stylization_build_model.py
@@ -20,6 +20,7 @@
 from magenta.models.image_stylization import ops
 import tensorflow.compat.v1 as tf
 import tf_slim as slim
+from tf_slim.nets import inception_v3
 
 
 def build_model(content_input_,
@@ -137,7 +138,7 @@ def style_prediction(style_input_,
           trainable=trainable):
         with slim.arg_scope(
             [slim.batch_norm, slim.dropout], is_training=is_training):
-          _, end_points = slim.nets.inception_v3.inception_v3_base(
+          _, end_points = inception_v3.inception_v3_base(
               style_input_,
               scope='InceptionV3',
               final_endpoint=inception_end_point)

diff --git a/magenta/version.py b/magenta/version.py
@@ -18,4 +18,4 @@
 pulling in all the dependencies in __init__.py.
 """
 
-__version__ = '2.0.2'
+__version__ = '2.0.3'