diff --git a/examples/4_Utils/tensorboard_advanced.py b/examples/4_Utils/tensorboard_advanced.py index 45a7f962..d07c266e 100644 --- a/examples/4_Utils/tensorboard_advanced.py +++ b/examples/4_Utils/tensorboard_advanced.py @@ -9,7 +9,7 @@ from __future__ import print_function -import tensorflow as tf +import tensorflow.compat.v1 as tf # Import MNIST data from tensorflow.examples.tutorials.mnist import input_data diff --git a/examples/4_Utils/tensorboard_basic.py b/examples/4_Utils/tensorboard_basic.py index 90f3f30c..cc5c76a2 100644 --- a/examples/4_Utils/tensorboard_basic.py +++ b/examples/4_Utils/tensorboard_basic.py @@ -11,11 +11,7 @@ import tensorflow as tf -# Import MNIST data -from tensorflow.examples.tutorials.mnist import input_data - -# !important 每个项目固定训练目录 /mnt/data -mnist = input_data.read_data_sets("/mnt/data/", one_hot=True) +mnist = tf.keras.datasets.mnist # Parameters learning_rate = 0.01