diff --git a/demo/word2vec/api_train_v2.py b/demo/word2vec/api_train_v2.py index 604adba192ee23aff4359df03d9f5e0ce21b8cde..8fea6feeef0c4de49e10e716769200f5e22d05ff 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 028f25a04676825a51530510abc579d48d2cbf4e..220d459525f0dccc58a56b149f841ca059bb5977 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'