We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fae3d8d commit 52e2eb6Copy full SHA for 52e2eb6
python/paddle/fluid/tests/book/high-level-api/fit_a_line/test_fit_a_line.py
@@ -38,7 +38,7 @@ def inference_program():
38
return y_predict
39
40
41
-def linear():
+def train_program():
42
y = fluid.layers.data(name='y', shape=[1], dtype='float32')
43
y_predict = inference_program()
44
@@ -104,7 +104,7 @@ def main(use_cuda):
104
# Directory for saving the trained model
105
params_dirname = "fit_a_line.inference.model"
106
107
- train(use_cuda, linear, params_dirname)
+ train(use_cuda, train_program, params_dirname)
108
infer(use_cuda, inference_program, params_dirname)
109
110
0 commit comments