From 18019e3c9eb32a33ca46daa68d84b8477402746f Mon Sep 17 00:00:00 2001 From: sfw134 <38463617+sfw134@users.noreply.github.com> Date: Wed, 13 Nov 2019 16:05:57 +0800 Subject: [PATCH 1/3] Update mnist.py --- docs/_static/code/zh/tools/save_and_restore/mnist.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/_static/code/zh/tools/save_and_restore/mnist.py b/docs/_static/code/zh/tools/save_and_restore/mnist.py index 0e663b27..cbb6e8c2 100644 --- a/docs/_static/code/zh/tools/save_and_restore/mnist.py +++ b/docs/_static/code/zh/tools/save_and_restore/mnist.py @@ -6,9 +6,9 @@ parser = argparse.ArgumentParser(description='Process some integers.') parser.add_argument('--mode', default='train', help='train or test') -parser.add_argument('--num_epochs', default=1) -parser.add_argument('--batch_size', default=50) -parser.add_argument('--learning_rate', default=0.001) +parser.add_argument('--num_epochs', default=1, type=int) +parser.add_argument('--batch_size', default=50, type=int) +parser.add_argument('--learning_rate', default=0.001, type=float) args = parser.parse_args() data_loader = MNISTLoader() From bf061743ca0698d2fa1136fe49681350d5ad5dee Mon Sep 17 00:00:00 2001 From: sfw134 <38463617+sfw134@users.noreply.github.com> Date: Wed, 13 Nov 2019 16:07:05 +0800 Subject: [PATCH 2/3] Update mnist_manager.py --- .../_static/code/zh/tools/save_and_restore/mnist_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py b/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py index ec2848a5..27876aa7 100644 --- a/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py +++ b/docs/_static/code/zh/tools/save_and_restore/mnist_manager.py @@ -6,9 +6,9 @@ parser = argparse.ArgumentParser(description='Process some integers.') parser.add_argument('--mode', default='train', help='train or test') -parser.add_argument('--num_epochs', default=1) -parser.add_argument('--batch_size', default=50) -parser.add_argument('--learning_rate', default=0.001) +parser.add_argument('--num_epochs', default=1, type=int) +parser.add_argument('--batch_size', default=50, type=int) +parser.add_argument('--learning_rate', default=0.001, type=float) args = parser.parse_args() data_loader = MNISTLoader() From 6cec0420bbc94733a7489aeaf8a81c2f8280dfb4 Mon Sep 17 00:00:00 2001 From: sfw134 <38463617+sfw134@users.noreply.github.com> Date: Fri, 15 Nov 2019 17:13:18 +0800 Subject: [PATCH 3/3] Update stylization.py --- docs/_static/code/zh/tfhub/stylization.py | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/_static/code/zh/tfhub/stylization.py b/docs/_static/code/zh/tfhub/stylization.py index c365d93d..833ac941 100644 --- a/docs/_static/code/zh/tfhub/stylization.py +++ b/docs/_static/code/zh/tfhub/stylization.py @@ -43,6 +43,7 @@ def load_image_local(image_path, image_size=(512, 512), preserve_aspect_ratio=Tr return img def show_image(image, title, save=False): + fig = plt.figure() plt.imshow(image, aspect='equal') plt.axis('off') if save: