提交 99036ce8 编写于 作者: W Wang,Jeff

update the image classification markdown

上级 2214bc99
...@@ -328,6 +328,15 @@ def train_program(): ...@@ -328,6 +328,15 @@ def train_program():
return [avg_cost, accuracy] return [avg_cost, accuracy]
``` ```
## Optimizer Function Configuration
In the following `Adam` optimizer, `learning_rate` specifies the learning rate in the optimization procedure.
```python
def optimizer_program():
return fluid.optimizer.Adam(learning_rate=0.001)
```
## Model Training ## Model Training
### Create Trainer ### Create Trainer
...@@ -340,7 +349,7 @@ use_cuda = False ...@@ -340,7 +349,7 @@ use_cuda = False
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
trainer = fluid.Trainer( trainer = fluid.Trainer(
train_func=train_program, train_func=train_program,
optimizer=fluid.optimizer.Adam(learning_rate=0.001), optimizer_func=optimizer_program,
place=place) place=place)
``` ```
......
...@@ -370,6 +370,15 @@ def train_program(): ...@@ -370,6 +370,15 @@ def train_program():
return [avg_cost, accuracy] return [avg_cost, accuracy]
``` ```
## Optimizer Function Configuration
In the following `Adam` optimizer, `learning_rate` specifies the learning rate in the optimization procedure.
```python
def optimizer_program():
return fluid.optimizer.Adam(learning_rate=0.001)
```
## Model Training ## Model Training
### Create Trainer ### Create Trainer
...@@ -382,7 +391,7 @@ use_cuda = False ...@@ -382,7 +391,7 @@ use_cuda = False
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
trainer = fluid.Trainer( trainer = fluid.Trainer(
train_func=train_program, train_func=train_program,
optimizer=fluid.optimizer.Adam(learning_rate=0.001), optimizer_func=optimizer_program,
place=place) place=place)
``` ```
......
...@@ -42,6 +42,10 @@ def train_network(): ...@@ -42,6 +42,10 @@ def train_network():
return [avg_cost, accuracy] return [avg_cost, accuracy]
def optimizer_program():
return fluid.optimizer.Adam(learning_rate=0.001)
def train(use_cuda, train_program, params_dirname): def train(use_cuda, train_program, params_dirname):
BATCH_SIZE = 128 BATCH_SIZE = 128
EPOCH_NUM = 2 EPOCH_NUM = 2
...@@ -74,9 +78,7 @@ def train(use_cuda, train_program, params_dirname): ...@@ -74,9 +78,7 @@ def train(use_cuda, train_program, params_dirname):
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
trainer = fluid.Trainer( trainer = fluid.Trainer(
train_func=train_program, train_func=train_program, optimizer_func=optimizer_program, place=place)
optimizer=fluid.optimizer.Adam(learning_rate=0.001),
place=place)
trainer.train( trainer.train(
reader=train_reader, reader=train_reader,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册