From 9f523515539ca0ff4faa8743297ac581811a94e9 Mon Sep 17 00:00:00 2001 From: Yancey1989 Date: Tue, 27 Jun 2017 15:59:11 +0800 Subject: [PATCH 1/2] save parameter --- demo/fit_a_line/train.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/demo/fit_a_line/train.py b/demo/fit_a_line/train.py index dbc6fbcc..7deee24a 100644 --- a/demo/fit_a_line/train.py +++ b/demo/fit_a_line/train.py @@ -1,5 +1,8 @@ import paddle.v2 as paddle import pcloud.dataset.uci_housing as uci_housing +import os +import gzip +trianer_id = os.getenv("PADDLE_INIT_TRAINER_ID") def main(): # init @@ -34,7 +37,10 @@ def event_handler(event): reader=paddle.batch(uci_housing.test(), batch_size=2), feeding=feeding) print "Test %d, Cost %f" % (event.pass_id, result.cost) - + if trainer_id == "0": + with gzip.open("fit-a-line_pass_%05d.tar.gz" % event.pass_id, + "w") as f: + parameters.to_tar(f) # training trainer.train( reader=paddle.batch( From f950d1366f26bd47fe3f5cb451c680bd2ef7bbef Mon Sep 17 00:00:00 2001 From: Yancey1989 Date: Wed, 28 Jun 2017 14:44:23 +0800 Subject: [PATCH 2/2] update --- demo/fit_a_line/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/fit_a_line/train.py b/demo/fit_a_line/train.py index 7deee24a..f27fdc16 100644 --- a/demo/fit_a_line/train.py +++ b/demo/fit_a_line/train.py @@ -2,7 +2,7 @@ import pcloud.dataset.uci_housing as uci_housing import os import gzip -trianer_id = os.getenv("PADDLE_INIT_TRAINER_ID") +trainer_id = os.getenv("PADDLE_INIT_TRAINER_ID") def main(): # init