提交 696312d8 authored 作者: Li's avatar Li 提交者: Frederic

picking works now

上级 a65ca93c
import unittest
import theano
import theano.tensor as T
import numpy
import cPickle
floatX = 'float32'
def test_pickle_unpickle():
x1 = T.fmatrix('x1')
x2 = T.fmatrix('x2')
x3 = theano.shared(numpy.ones((10,10),dtype=floatX))
y = T.sum(T.sum(x1**2+x2) + x3)
f = theano.function([x1,x2],y)
pkl_path = open('thean_fn.pkl','wb')
cPickle.dump(f, pkl_path, -1)
pkl_path = open('thean_fn.pkl','r')
f_ = cPickle.load(pkl_path)
in1 = numpy.ones((10,10), dtype=floatX)
in2 = numpy.ones((10,10), dtype=floatX)
assert f(in1,in2) == f_(in1,in2)
print f(in1,in2)
if __name__ == '__main__':
test_pickle_unpickle()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论