From 35f1dfde72f9be8d9fccdaaf5d328da9f8ffcc36 Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Mon, 24 Apr 2017 13:58:14 +0800
Subject: [PATCH] chage trainer.save_parameter to trainer.save_parameter_to_tar

---
 demo/word2vec/api_train_v2.py | 4 ++--
 python/paddle/v2/trainer.py   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/demo/word2vec/api_train_v2.py b/demo/word2vec/api_train_v2.py
index 604adba19..8fea6feee 100644
--- a/demo/word2vec/api_train_v2.py
+++ b/demo/word2vec/api_train_v2.py
@@ -29,7 +29,7 @@ def main():
     else:
         paddle.init(
             use_gpu=False,
-            trainer_count=1,
+            trainer_count=2,
             port=7164,
             ports_num=1,
             ports_num_for_sparse=1,
@@ -69,7 +69,7 @@ def main():
     def event_handler(event):
         if isinstance(event, paddle.event.EndIteration):
             if event.batch_id % 100 == 0:
-                trainer.save_parameter("output", "batch-" + str(event.batch_id))
+                trainer.save_parameter_to_tar("output", "batch-" + str(event.batch_id))
                 result = trainer.test(
                     paddle.batch(
                         paddle.dataset.imikolov.test(word_dict, N), 32))
diff --git a/python/paddle/v2/trainer.py b/python/paddle/v2/trainer.py
index 028f25a04..220d45952 100644
--- a/python/paddle/v2/trainer.py
+++ b/python/paddle/v2/trainer.py
@@ -98,7 +98,7 @@ class SGD(object):
             self.__gradient_machine__.prefetch(in_args)
             self.__parameter_updater__.getParametersRemote()
 
-    def save_parameter(self, dir_name, file_name):
+    def save_parameter_to_tar(self, dir_name, file_name):
         if not os.path.exists(dir_name):
             os.makedirs(dir_name)
         param_file_name = dir_name + "/" + file_name + '.tar.gz'
-- 
GitLab