chage trainer.save_parameter to trainer.save_parameter_to_tar

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))

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'