parent
72bcf7b368
commit
f5dc9eadaa
@ -0,0 +1,14 @@
|
||||
import paddle.v2 as paddle
|
||||
import numpy as np
|
||||
|
||||
paddle.init(use_gpu=False)
|
||||
x = paddle.layer.data(name='x', type=paddle.data_type.dense_vector(2))
|
||||
y_predict = paddle.layer.fc(input=x, size=1, act=paddle.activation.Linear())
|
||||
|
||||
# loading the model which generated by training
|
||||
with open('hello_params_pass_90.tar', 'r') as f:
|
||||
parameters = paddle.parameters.Parameters.from_tar(f)
|
||||
|
||||
i = [[[1, 2]]]
|
||||
|
||||
print paddle.infer(output_layer=y_predict, parameters=parameters, input=i)
|
Loading…
Reference in new issue