Skip to content

Commit 9b41b08

Browse files
committed
Remove unnecessary import in api_train.py
1 parent 763a30f commit 9b41b08

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

demo/mnist/api_train.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
import numpy as np
1313
import random
1414
from mnist_util import read_from_mnist
15-
import paddle.trainer_config_helpers.config_parser_utils as config_parser_utils
1615
from paddle.trainer_config_helpers import *
1716

1817

@@ -80,14 +79,13 @@ def main():
8079
# enable_types = [value, gradient, momentum, etc]
8180
# For each optimizer(SGD, Adam), GradientMachine should enable different
8281
# buffers.
83-
opt_config_proto = config_parser_utils.parse_optimizer_config(
84-
optimizer_config)
82+
opt_config_proto = parse_optimizer_config(optimizer_config)
8583
opt_config = api.OptimizationConfig.createFromProto(opt_config_proto)
8684
_temp_optimizer_ = api.ParameterOptimizer.create(opt_config)
8785
enable_types = _temp_optimizer_.getParameterTypes()
8886

8987
# Create Simple Gradient Machine.
90-
model_config = config_parser_utils.parse_network_config(network_config)
88+
model_config = parse_network_config(network_config)
9189
m = api.GradientMachine.createFromConfigProto(
9290
model_config, api.CREATE_MODE_NORMAL, enable_types)
9391

0 commit comments

Comments
 (0)