diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index 0ad55ceb528258a20a196ce441e675da5973fa72..adc6c898240601f9b7ecf81f726d4465c39d92b2 100755 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -35,10 +35,10 @@ build:server: paths: - build/third_party script: - #- export http_proxy=http://172.19.57.45:3128 - #- export https_proxy=http://172.19.57.45:3128 - - export http_proxy=http://agent.baidu.com:8118 - - export https_proxy=http://agent.baidu.com:8118 + - export http_proxy=http://172.19.57.45:3128 + - export https_proxy=http://172.19.57.45:3128 + #- export http_proxy=http://agent.baidu.com:8118 + #- export https_proxy=http://agent.baidu.com:8118 - mkdir -p build - cd build - ../paddle/fluid/lite/tools/build.sh cmake_x86 diff --git a/paddle/fluid/lite/core/naive_test_model.py b/paddle/fluid/lite/core/naive_test_model.py index 832661e5ee86f2759acfeb4a6a410cce6050ad53..f89a5e115fa805bab818cabeab1d63cac00158d0 100644 --- a/paddle/fluid/lite/core/naive_test_model.py +++ b/paddle/fluid/lite/core/naive_test_model.py @@ -18,10 +18,10 @@ import numpy as np import paddle.fluid as fluid from paddle.fluid.backward import append_backward -a = fluid.layers.data(name="a", shape=[100], dtype='float32') -label = fluid.layers.data(name="label", shape=[100], dtype='float32') +a = fluid.layers.data(name="a", shape=[2], dtype='float32') +label = fluid.layers.data(name="label", shape=[10], dtype='float32') -a1 = fluid.layers.fc(input=a, size=500, act=None, bias_attr=False) +a1 = fluid.layers.fc(input=a, size=3, act=None, bias_attr=False) cost = fluid.layers.square_error_cost(a1, label) avg_cost = fluid.layers.mean(cost) @@ -36,7 +36,7 @@ exe.run(fluid.default_startup_program()) with open('startup_program.pb', 'wb') as f: f.write(fluid.default_startup_program().desc.serialize_to_string()) -data_1 = np.array(numpy.random.random([100, 100]), dtype='float32') +#data_1 = np.array(numpy.random.random([100, 100]), dtype='float32') #fluid.default_main_program().desc. @@ -50,7 +50,7 @@ with open('main_program.pb', 'wb') as f: #outs = exe.run(program=prog, feed={'a':data_1, }, fetch_list=[cost]) -sys.exit(0) +#sys.exit(0) fluid.io.save_inference_model("./model2", [a.name], [a1], exe) -print(numpy.array(outs)) +#print(numpy.array(outs))