From 1a9500e6e2228e49123405ff3d91f8cb4703bf9a Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Sun, 11 Feb 2018 10:47:48 +0800 Subject: [PATCH] Rename optimizer --- fluid/DeepASR/tools/profile.py | 4 ++-- fluid/DeepASR/train.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/fluid/DeepASR/tools/profile.py b/fluid/DeepASR/tools/profile.py index de8c27a8..492a3aa2 100644 --- a/fluid/DeepASR/tools/profile.py +++ b/fluid/DeepASR/tools/profile.py @@ -125,9 +125,9 @@ def profile(args): class_num=1749, parallel=args.parallel) - adam_optimizer = fluid.optimizer.Momentum( + optimizer = fluid.optimizer.Momentum( learning_rate=args.learning_rate, momentum=0.9) - adam_optimizer.minimize(avg_cost) + optimizer.minimize(avg_cost) place = fluid.CPUPlace() if args.device == 'CPU' else fluid.CUDAPlace(0) exe = fluid.Executor(place) diff --git a/fluid/DeepASR/train.py b/fluid/DeepASR/train.py index 85a5b8d1..1ef97acd 100644 --- a/fluid/DeepASR/train.py +++ b/fluid/DeepASR/train.py @@ -143,9 +143,9 @@ def train(args): class_num=1749, parallel=args.parallel) - adam_optimizer = fluid.optimizer.Momentum( + optimizer = fluid.optimizer.Momentum( learning_rate=args.learning_rate, momentum=0.9) - adam_optimizer.minimize(avg_cost) + optimizer.minimize(avg_cost) # program for test test_program = fluid.default_main_program().clone() -- GitLab