提交 848efd5f 编写于 作者: H helinwang 提交者: GitHub

Merge pull request #342 from helinwang/gzip

Make same model examples no longer compress model.
...@@ -132,7 +132,6 @@ PaddlePaddle在API中提供了自动加载[MNIST](http://yann.lecun.com/exdb/mni ...@@ -132,7 +132,6 @@ PaddlePaddle在API中提供了自动加载[MNIST](http://yann.lecun.com/exdb/mni
首先,加载PaddlePaddle的V2 api包。 首先,加载PaddlePaddle的V2 api包。
```python ```python
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
``` ```
其次,定义三个不同的分类器: 其次,定义三个不同的分类器:
...@@ -256,7 +255,7 @@ def event_handler_plot(event): ...@@ -256,7 +255,7 @@ def event_handler_plot(event):
step += 1 step += 1
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
...@@ -275,7 +274,7 @@ def event_handler(event): ...@@ -275,7 +274,7 @@ def event_handler(event):
event.pass_id, event.batch_id, event.cost, event.metrics) event.pass_id, event.batch_id, event.cost, event.metrics)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
......
...@@ -131,7 +131,6 @@ PaddlePaddle provides a Python module, `paddle.dataset.mnist`, which downloads a ...@@ -131,7 +131,6 @@ PaddlePaddle provides a Python module, `paddle.dataset.mnist`, which downloads a
A PaddlePaddle program starts from importing the API package: A PaddlePaddle program starts from importing the API package:
```python ```python
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
``` ```
...@@ -251,7 +250,7 @@ def event_handler_plot(event): ...@@ -251,7 +250,7 @@ def event_handler_plot(event):
step += 1 step += 1
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
...@@ -271,7 +270,7 @@ def event_handler(event): ...@@ -271,7 +270,7 @@ def event_handler(event):
event.pass_id, event.batch_id, event.cost, event.metrics) event.pass_id, event.batch_id, event.cost, event.metrics)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
......
...@@ -174,7 +174,6 @@ PaddlePaddle在API中提供了自动加载[MNIST](http://yann.lecun.com/exdb/mni ...@@ -174,7 +174,6 @@ PaddlePaddle在API中提供了自动加载[MNIST](http://yann.lecun.com/exdb/mni
首先,加载PaddlePaddle的V2 api包。 首先,加载PaddlePaddle的V2 api包。
```python ```python
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
``` ```
其次,定义三个不同的分类器: 其次,定义三个不同的分类器:
...@@ -298,7 +297,7 @@ def event_handler_plot(event): ...@@ -298,7 +297,7 @@ def event_handler_plot(event):
step += 1 step += 1
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
...@@ -317,7 +316,7 @@ def event_handler(event): ...@@ -317,7 +316,7 @@ def event_handler(event):
event.pass_id, event.batch_id, event.cost, event.metrics) event.pass_id, event.batch_id, event.cost, event.metrics)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
......
...@@ -173,7 +173,6 @@ PaddlePaddle provides a Python module, `paddle.dataset.mnist`, which downloads a ...@@ -173,7 +173,6 @@ PaddlePaddle provides a Python module, `paddle.dataset.mnist`, which downloads a
A PaddlePaddle program starts from importing the API package: A PaddlePaddle program starts from importing the API package:
```python ```python
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
``` ```
...@@ -293,7 +292,7 @@ def event_handler_plot(event): ...@@ -293,7 +292,7 @@ def event_handler_plot(event):
step += 1 step += 1
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
...@@ -313,7 +312,7 @@ def event_handler(event): ...@@ -313,7 +312,7 @@ def event_handler(event):
event.pass_id, event.batch_id, event.cost, event.metrics) event.pass_id, event.batch_id, event.cost, event.metrics)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
......
import gzip
import os import os
from PIL import Image from PIL import Image
import numpy as np import numpy as np
...@@ -85,7 +84,7 @@ def main(): ...@@ -85,7 +84,7 @@ def main():
event.pass_id, event.batch_id, event.cost, event.metrics) event.pass_id, event.batch_id, event.cost, event.metrics)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=paddle.batch( result = trainer.test(reader=paddle.batch(
......
...@@ -156,7 +156,6 @@ Paddle API提供了自动加载cifar数据集模块 `paddle.dataset.cifar`。 ...@@ -156,7 +156,6 @@ Paddle API提供了自动加载cifar数据集模块 `paddle.dataset.cifar`。
```python ```python
import sys import sys
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
from vgg import vgg_bn_drop from vgg import vgg_bn_drop
from resnet import resnet_cifar10 from resnet import resnet_cifar10
...@@ -431,7 +430,7 @@ def event_handler(event): ...@@ -431,7 +430,7 @@ def event_handler(event):
sys.stdout.flush() sys.stdout.flush()
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test( result = trainer.test(
...@@ -498,7 +497,7 @@ test_data = [] ...@@ -498,7 +497,7 @@ test_data = []
cur_dir = os.path.dirname(os.path.realpath(__file__)) cur_dir = os.path.dirname(os.path.realpath(__file__))
test_data.append((load_image(cur_dir + '/image/dog.png'),)) test_data.append((load_image(cur_dir + '/image/dog.png'),))
# with gzip.open('params_pass_50.tar.gz', 'r') as f: # with open('params_pass_50.tar', 'r') as f:
# parameters = paddle.parameters.Parameters.from_tar(f) # parameters = paddle.parameters.Parameters.from_tar(f)
probs = paddle.infer( probs = paddle.infer(
......
...@@ -169,7 +169,6 @@ We must import and initialize PaddlePaddle (enable/disable GPU, set the number o ...@@ -169,7 +169,6 @@ We must import and initialize PaddlePaddle (enable/disable GPU, set the number o
```python ```python
import sys import sys
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
from vgg import vgg_bn_drop from vgg import vgg_bn_drop
from resnet import resnet_cifar10 from resnet import resnet_cifar10
...@@ -438,7 +437,7 @@ def event_handler(event): ...@@ -438,7 +437,7 @@ def event_handler(event):
sys.stdout.flush() sys.stdout.flush()
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test( result = trainer.test(
...@@ -508,7 +507,7 @@ cur_dir = os.path.dirname(os.path.realpath(__file__)) ...@@ -508,7 +507,7 @@ cur_dir = os.path.dirname(os.path.realpath(__file__))
test_data.append((load_image(cur_dir + '/image/dog.png'),)) test_data.append((load_image(cur_dir + '/image/dog.png'),))
# users can remove the comments and change the model name # users can remove the comments and change the model name
# with gzip.open('params_pass_50.tar.gz', 'r') as f: # with open('params_pass_50.tar', 'r') as f:
# parameters = paddle.parameters.Parameters.from_tar(f) # parameters = paddle.parameters.Parameters.from_tar(f)
probs = paddle.infer( probs = paddle.infer(
......
...@@ -198,7 +198,6 @@ Paddle API提供了自动加载cifar数据集模块 `paddle.dataset.cifar`。 ...@@ -198,7 +198,6 @@ Paddle API提供了自动加载cifar数据集模块 `paddle.dataset.cifar`。
```python ```python
import sys import sys
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
from vgg import vgg_bn_drop from vgg import vgg_bn_drop
from resnet import resnet_cifar10 from resnet import resnet_cifar10
...@@ -473,7 +472,7 @@ def event_handler(event): ...@@ -473,7 +472,7 @@ def event_handler(event):
sys.stdout.flush() sys.stdout.flush()
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test( result = trainer.test(
...@@ -540,7 +539,7 @@ test_data = [] ...@@ -540,7 +539,7 @@ test_data = []
cur_dir = os.path.dirname(os.path.realpath(__file__)) cur_dir = os.path.dirname(os.path.realpath(__file__))
test_data.append((load_image(cur_dir + '/image/dog.png'),)) test_data.append((load_image(cur_dir + '/image/dog.png'),))
# with gzip.open('params_pass_50.tar.gz', 'r') as f: # with open('params_pass_50.tar', 'r') as f:
# parameters = paddle.parameters.Parameters.from_tar(f) # parameters = paddle.parameters.Parameters.from_tar(f)
probs = paddle.infer( probs = paddle.infer(
......
...@@ -211,7 +211,6 @@ We must import and initialize PaddlePaddle (enable/disable GPU, set the number o ...@@ -211,7 +211,6 @@ We must import and initialize PaddlePaddle (enable/disable GPU, set the number o
```python ```python
import sys import sys
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
from vgg import vgg_bn_drop from vgg import vgg_bn_drop
from resnet import resnet_cifar10 from resnet import resnet_cifar10
...@@ -480,7 +479,7 @@ def event_handler(event): ...@@ -480,7 +479,7 @@ def event_handler(event):
sys.stdout.flush() sys.stdout.flush()
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test( result = trainer.test(
...@@ -550,7 +549,7 @@ cur_dir = os.path.dirname(os.path.realpath(__file__)) ...@@ -550,7 +549,7 @@ cur_dir = os.path.dirname(os.path.realpath(__file__))
test_data.append((load_image(cur_dir + '/image/dog.png'),)) test_data.append((load_image(cur_dir + '/image/dog.png'),))
# users can remove the comments and change the model name # users can remove the comments and change the model name
# with gzip.open('params_pass_50.tar.gz', 'r') as f: # with open('params_pass_50.tar', 'r') as f:
# parameters = paddle.parameters.Parameters.from_tar(f) # parameters = paddle.parameters.Parameters.from_tar(f)
probs = paddle.infer( probs = paddle.infer(
......
...@@ -13,7 +13,6 @@ ...@@ -13,7 +13,6 @@
# limitations under the License # limitations under the License
import sys import sys
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
...@@ -67,7 +66,7 @@ def main(): ...@@ -67,7 +66,7 @@ def main():
sys.stdout.flush() sys.stdout.flush()
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test( result = trainer.test(
...@@ -116,7 +115,7 @@ def main(): ...@@ -116,7 +115,7 @@ def main():
test_data.append((load_image(cur_dir + '/image/dog.png'), )) test_data.append((load_image(cur_dir + '/image/dog.png'), ))
# users can remove the comments and change the model name # users can remove the comments and change the model name
# with gzip.open('params_pass_50.tar.gz', 'r') as f: # with open('params_pass_50.tar', 'r') as f:
# parameters = paddle.parameters.Parameters.from_tar(f) # parameters = paddle.parameters.Parameters.from_tar(f)
probs = paddle.infer( probs = paddle.infer(
......
...@@ -302,8 +302,6 @@ trainer = paddle.trainer.SGD(cost, parameters, adagrad) ...@@ -302,8 +302,6 @@ trainer = paddle.trainer.SGD(cost, parameters, adagrad)
`paddle.batch`的输入是一个reader,输出是一个batched reader —— 在PaddlePaddle里,一个reader每次yield一条训练数据,而一个batched reader每次yield一个minbatch。 `paddle.batch`的输入是一个reader,输出是一个batched reader —— 在PaddlePaddle里,一个reader每次yield一条训练数据,而一个batched reader每次yield一个minbatch。
```python ```python
import gzip
def event_handler(event): def event_handler(event):
if isinstance(event, paddle.event.EndIteration): if isinstance(event, paddle.event.EndIteration):
if event.batch_id % 100 == 0: if event.batch_id % 100 == 0:
...@@ -315,7 +313,7 @@ def event_handler(event): ...@@ -315,7 +313,7 @@ def event_handler(event):
paddle.batch( paddle.batch(
paddle.dataset.imikolov.test(word_dict, N), 32)) paddle.dataset.imikolov.test(word_dict, N), 32))
print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics) print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics)
with gzip.open("model_%d.tar.gz"%event.pass_id, 'w') as f: with open("model_%d.tar"%event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
trainer.train( trainer.train(
......
...@@ -313,8 +313,6 @@ Next, we will begin the training process. `paddle.dataset.imikolov.train()` and ...@@ -313,8 +313,6 @@ Next, we will begin the training process. `paddle.dataset.imikolov.train()` and
`paddle.batch` takes reader as input, outputs a **batched reader**: In PaddlePaddle, a reader outputs a single data instance at a time but batched reader outputs a minibatch of data instances. `paddle.batch` takes reader as input, outputs a **batched reader**: In PaddlePaddle, a reader outputs a single data instance at a time but batched reader outputs a minibatch of data instances.
```python ```python
import gzip
def event_handler(event): def event_handler(event):
if isinstance(event, paddle.event.EndIteration): if isinstance(event, paddle.event.EndIteration):
if event.batch_id % 100 == 0: if event.batch_id % 100 == 0:
...@@ -326,7 +324,7 @@ def event_handler(event): ...@@ -326,7 +324,7 @@ def event_handler(event):
paddle.batch( paddle.batch(
paddle.dataset.imikolov.test(word_dict, N), 32)) paddle.dataset.imikolov.test(word_dict, N), 32))
print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics) print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics)
with gzip.open("model_%d.tar.gz"%event.pass_id, 'w') as f: with open("model_%d.tar"%event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
trainer.train( trainer.train(
......
...@@ -344,8 +344,6 @@ trainer = paddle.trainer.SGD(cost, parameters, adagrad) ...@@ -344,8 +344,6 @@ trainer = paddle.trainer.SGD(cost, parameters, adagrad)
`paddle.batch`的输入是一个reader,输出是一个batched reader —— 在PaddlePaddle里,一个reader每次yield一条训练数据,而一个batched reader每次yield一个minbatch。 `paddle.batch`的输入是一个reader,输出是一个batched reader —— 在PaddlePaddle里,一个reader每次yield一条训练数据,而一个batched reader每次yield一个minbatch。
```python ```python
import gzip
def event_handler(event): def event_handler(event):
if isinstance(event, paddle.event.EndIteration): if isinstance(event, paddle.event.EndIteration):
if event.batch_id % 100 == 0: if event.batch_id % 100 == 0:
...@@ -357,7 +355,7 @@ def event_handler(event): ...@@ -357,7 +355,7 @@ def event_handler(event):
paddle.batch( paddle.batch(
paddle.dataset.imikolov.test(word_dict, N), 32)) paddle.dataset.imikolov.test(word_dict, N), 32))
print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics) print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics)
with gzip.open("model_%d.tar.gz"%event.pass_id, 'w') as f: with open("model_%d.tar"%event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
trainer.train( trainer.train(
......
...@@ -355,8 +355,6 @@ Next, we will begin the training process. `paddle.dataset.imikolov.train()` and ...@@ -355,8 +355,6 @@ Next, we will begin the training process. `paddle.dataset.imikolov.train()` and
`paddle.batch` takes reader as input, outputs a **batched reader**: In PaddlePaddle, a reader outputs a single data instance at a time but batched reader outputs a minibatch of data instances. `paddle.batch` takes reader as input, outputs a **batched reader**: In PaddlePaddle, a reader outputs a single data instance at a time but batched reader outputs a minibatch of data instances.
```python ```python
import gzip
def event_handler(event): def event_handler(event):
if isinstance(event, paddle.event.EndIteration): if isinstance(event, paddle.event.EndIteration):
if event.batch_id % 100 == 0: if event.batch_id % 100 == 0:
...@@ -368,7 +366,7 @@ def event_handler(event): ...@@ -368,7 +366,7 @@ def event_handler(event):
paddle.batch( paddle.batch(
paddle.dataset.imikolov.test(word_dict, N), 32)) paddle.dataset.imikolov.test(word_dict, N), 32))
print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics) print "Pass %d, Testing metrics %s" % (event.pass_id, result.metrics)
with gzip.open("model_%d.tar.gz"%event.pass_id, 'w') as f: with open("model_%d.tar"%event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
trainer.train( trainer.train(
......
...@@ -189,7 +189,6 @@ conll05st-release/ ...@@ -189,7 +189,6 @@ conll05st-release/
```python ```python
import math import math
import numpy as np import numpy as np
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
import paddle.v2.dataset.conll05 as conll05 import paddle.v2.dataset.conll05 as conll05
import paddle.v2.evaluator as evaluator import paddle.v2.evaluator as evaluator
...@@ -448,7 +447,7 @@ def event_handler(event): ...@@ -448,7 +447,7 @@ def event_handler(event):
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=reader, feeding=feeding) result = trainer.test(reader=reader, feeding=feeding)
......
...@@ -211,7 +211,6 @@ Here we fetch the dictionary, and print its size: ...@@ -211,7 +211,6 @@ Here we fetch the dictionary, and print its size:
```python ```python
import math import math
import numpy as np import numpy as np
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
import paddle.v2.dataset.conll05 as conll05 import paddle.v2.dataset.conll05 as conll05
import paddle.v2.evaluator as evaluator import paddle.v2.evaluator as evaluator
...@@ -466,7 +465,7 @@ def event_handler(event): ...@@ -466,7 +465,7 @@ def event_handler(event):
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=reader, feeding=feeding) result = trainer.test(reader=reader, feeding=feeding)
......
...@@ -231,7 +231,6 @@ conll05st-release/ ...@@ -231,7 +231,6 @@ conll05st-release/
```python ```python
import math import math
import numpy as np import numpy as np
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
import paddle.v2.dataset.conll05 as conll05 import paddle.v2.dataset.conll05 as conll05
import paddle.v2.evaluator as evaluator import paddle.v2.evaluator as evaluator
...@@ -490,7 +489,7 @@ def event_handler(event): ...@@ -490,7 +489,7 @@ def event_handler(event):
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=reader, feeding=feeding) result = trainer.test(reader=reader, feeding=feeding)
......
...@@ -253,7 +253,6 @@ Here we fetch the dictionary, and print its size: ...@@ -253,7 +253,6 @@ Here we fetch the dictionary, and print its size:
```python ```python
import math import math
import numpy as np import numpy as np
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
import paddle.v2.dataset.conll05 as conll05 import paddle.v2.dataset.conll05 as conll05
import paddle.v2.evaluator as evaluator import paddle.v2.evaluator as evaluator
...@@ -508,7 +507,7 @@ def event_handler(event): ...@@ -508,7 +507,7 @@ def event_handler(event):
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=reader, feeding=feeding) result = trainer.test(reader=reader, feeding=feeding)
......
import math import math
import numpy as np import numpy as np
import gzip
import paddle.v2 as paddle import paddle.v2 as paddle
import paddle.v2.dataset.conll05 as conll05 import paddle.v2.dataset.conll05 as conll05
import paddle.v2.evaluator as evaluator import paddle.v2.evaluator as evaluator
...@@ -183,7 +182,7 @@ def main(): ...@@ -183,7 +182,7 @@ def main():
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
with gzip.open('params_pass_%d.tar.gz' % event.pass_id, 'w') as f: with open('params_pass_%d.tar' % event.pass_id, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
result = trainer.test(reader=reader, feeding=feeding) result = trainer.test(reader=reader, feeding=feeding)
......
import sys import sys
import gzip
import numpy as np import numpy as np
import paddle.v2 as paddle import paddle.v2 as paddle
def save_model(parameters, save_path): def save_model(parameters, save_path):
with gzip.open(save_path, 'w') as f: with open(save_path, 'w') as f:
parameters.to_tar(f) parameters.to_tar(f)
...@@ -173,13 +172,13 @@ def main(): ...@@ -173,13 +172,13 @@ def main():
sys.stdout.flush() sys.stdout.flush()
if not event.batch_id % 10: if not event.batch_id % 10:
save_path = 'params_pass_%05d_batch_%05d.tar.gz' % ( save_path = 'params_pass_%05d_batch_%05d.tar' % (
event.pass_id, event.batch_id) event.pass_id, event.batch_id)
save_model(parameters, save_path) save_model(parameters, save_path)
if isinstance(event, paddle.event.EndPass): if isinstance(event, paddle.event.EndPass):
# save parameters # save parameters
save_path = 'params_pass_%05d.tar.gz' % (event.pass_id) save_path = 'params_pass_%05d.tar' % (event.pass_id)
save_model(parameters, save_path) save_model(parameters, save_path)
# start to train # start to train
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册