提交 3d2268e7 编写于 作者: R ruri 提交者: lvmengsi

Fix predict bug in image classification (#833)

* fix typo:Infererence

* fix predict bug in ic
上级 b5d917eb
......@@ -338,7 +338,7 @@ def train_program():
cost = fluid.layers.cross_entropy(input=predict, label=label)
avg_cost = fluid.layers.mean(cost)
accuracy = fluid.layers.accuracy(input=predict, label=label)
return [avg_cost, accuracy]
return [avg_cost, accuracy, predict]
```
## Optimizer Function 配置
......@@ -383,7 +383,7 @@ feed_order = ['pixel', 'label']
main_program = fluid.default_main_program()
star_program = fluid.default_startup_program()
avg_cost, acc = train_program()
avg_cost, acc, predict = train_program()
# Test program
test_program = main_program.clone(for_test=True)
......
......@@ -335,7 +335,7 @@ def train_program():
cost = fluid.layers.cross_entropy(input=predict, label=label)
avg_cost = fluid.layers.mean(cost)
accuracy = fluid.layers.accuracy(input=predict, label=label)
return [avg_cost, accuracy]
return [avg_cost, accuracy, predict]
```
## Optimizer Function Configuration
......@@ -383,7 +383,7 @@ feed_order = ['pixel', 'label']
main_program = fluid.default_main_program()
star_program = fluid.default_startup_program()
avg_cost, acc = train_program()
avg_cost, acc, predict = train_program()
# Test program
test_program = main_program.clone(for_test=True)
......
......@@ -380,7 +380,7 @@ def train_program():
cost = fluid.layers.cross_entropy(input=predict, label=label)
avg_cost = fluid.layers.mean(cost)
accuracy = fluid.layers.accuracy(input=predict, label=label)
return [avg_cost, accuracy]
return [avg_cost, accuracy, predict]
```
## Optimizer Function 配置
......@@ -425,7 +425,7 @@ feed_order = ['pixel', 'label']
main_program = fluid.default_main_program()
star_program = fluid.default_startup_program()
avg_cost, acc = train_program()
avg_cost, acc, predict = train_program()
# Test program
test_program = main_program.clone(for_test=True)
......
......@@ -377,7 +377,7 @@ def train_program():
cost = fluid.layers.cross_entropy(input=predict, label=label)
avg_cost = fluid.layers.mean(cost)
accuracy = fluid.layers.accuracy(input=predict, label=label)
return [avg_cost, accuracy]
return [avg_cost, accuracy, predict]
```
## Optimizer Function Configuration
......@@ -425,7 +425,7 @@ feed_order = ['pixel', 'label']
main_program = fluid.default_main_program()
star_program = fluid.default_startup_program()
avg_cost, acc = train_program()
avg_cost, acc, predict = train_program()
# Test program
test_program = main_program.clone(for_test=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册