File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 12
12
import numpy as np
13
13
import random
14
14
from mnist_util import read_from_mnist
15
- import paddle .trainer_config_helpers .config_parser_utils as config_parser_utils
16
15
from paddle .trainer_config_helpers import *
17
16
18
17
@@ -80,14 +79,13 @@ def main():
80
79
# enable_types = [value, gradient, momentum, etc]
81
80
# For each optimizer(SGD, Adam), GradientMachine should enable different
82
81
# buffers.
83
- opt_config_proto = config_parser_utils .parse_optimizer_config (
84
- optimizer_config )
82
+ opt_config_proto = parse_optimizer_config (optimizer_config )
85
83
opt_config = api .OptimizationConfig .createFromProto (opt_config_proto )
86
84
_temp_optimizer_ = api .ParameterOptimizer .create (opt_config )
87
85
enable_types = _temp_optimizer_ .getParameterTypes ()
88
86
89
87
# Create Simple Gradient Machine.
90
- model_config = config_parser_utils . parse_network_config (network_config )
88
+ model_config = parse_network_config (network_config )
91
89
m = api .GradientMachine .createFromConfigProto (
92
90
model_config , api .CREATE_MODE_NORMAL , enable_types )
93
91
You can’t perform that action at this time.
0 commit comments