diff --git a/paddle/go/pserver/optimizer.c b/paddle/go/pserver/optimizer.c index 36a612a56f..8d63089b4c 100644 --- a/paddle/go/pserver/optimizer.c +++ b/paddle/go/pserver/optimizer.c @@ -36,11 +36,11 @@ void release_SGD(void *optimizer) { } paddle_optimizer* paddle_create_SGD_optimizer(double learning_rate) { - SGD_optimizer* o = (SGD_optimizer*)malloc(sizeof(SGD_optimizer)); - o->learning_rate = learning_rate; - paddle_optimizer* container = (paddle_optimizer*)malloc(sizeof(paddle_optimizer)); - container->update = update_SGD; - container->release = release_SGD; - container->optimizer = o; - return container; + SGD_optimizer* impl = (SGD_optimizer*)malloc(sizeof(SGD_optimizer)); + impl->learning_rate = learning_rate; + paddle_optimizer* opt = (paddle_optimizer*)malloc(sizeof(paddle_optimizer)); + opt->update = update_SGD; + opt->release = release_SGD; + opt->optimizer = impl; + return opt; }