From 8d17b08a4360d40556c6fc37ea5460085fa55941 Mon Sep 17 00:00:00 2001 From: niboshi Date: Wed, 7 Aug 2019 13:24:11 +0900 Subject: [PATCH] Fix static_graph docs code examples --- docs/source/reference/static_graph.rst | 8 ++++++-- examples/static_graph_optimizations/mnist/train_mnist.py | 6 ++++++ 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/docs/source/reference/static_graph.rst b/docs/source/reference/static_graph.rst index b5a6265b74cc..3c9ae83bfbe4 100644 --- a/docs/source/reference/static_graph.rst +++ b/docs/source/reference/static_graph.rst @@ -28,7 +28,9 @@ The first step is to import the necessary packages: .. literalinclude:: ../../../examples/static_graph_optimizations/mnist/train_mnist.py :language: python - :lines: 15-16 + :linenos: + :lines: 24-25 + :lineno-start: 24 :caption: train_mnist.py Since the neural network model ``MLP`` corresponds to a static graph, we can annotate it as a static graph by @@ -39,7 +41,9 @@ the documentation. .. literalinclude:: ../../../examples/static_graph_optimizations/mnist/train_mnist.py :language: python - :lines: 22-41 + :linenos: + :lines: 34-53 + :lineno-start: 34 :caption: train_mnist.py .. note:: diff --git a/examples/static_graph_optimizations/mnist/train_mnist.py b/examples/static_graph_optimizations/mnist/train_mnist.py index 5ec3c4d217d9..25fc7277b22b 100755 --- a/examples/static_graph_optimizations/mnist/train_mnist.py +++ b/examples/static_graph_optimizations/mnist/train_mnist.py @@ -4,7 +4,13 @@ to support the static subgraph optimizations feature. Note that the code is mostly unchanged except for the addition of the `@static_graph` decorator to the model chain's `__call__()` method. + +Note for contributors: +This example code is referred to from the documentation. +If this file is to be modified, please also update the line numbers in +`docs/source/reference/static_graph.rst` accordingly. """ + from __future__ import print_function import argparse