diff --git a/paddle/api/ParameterUpdater.cpp b/paddle/api/ParameterUpdater.cpp index e84bb63866094d004c03e4c8c0c4838c99c551e6..33aa11850cc5970d11163eb3f4d4a18891e20da7 100644 --- a/paddle/api/ParameterUpdater.cpp +++ b/paddle/api/ParameterUpdater.cpp @@ -32,8 +32,7 @@ ParameterUpdater *ParameterUpdater::createRemoteUpdater( OptimizationConfig *config, int passCount) { auto updater = new ParameterUpdater(); std::unique_ptr localUpdater; - localUpdater.reset(new paddle::SgdThreadUpdater(config->m->getConfig())); - updater->m->updater.reset(new paddle::ConcurrentRemoteParameterUpdater( + updater->m->updater.reset(new paddle::RemoteParameterUpdater( config->m->getConfig(), passCount, std::move(localUpdater))); return updater; }