提交 c9ffe855 编写于 作者: L liuruilong

change test model file path

上级 dd41e2cb
...@@ -23,7 +23,6 @@ ...@@ -23,7 +23,6 @@
# Compiled Static libraries # Compiled Static libraries
*.lai *.lai
*.la *.la
*.a
*.lib *.lib
# Executables # Executables
...@@ -52,3 +51,6 @@ paddle-mobile.cbp ...@@ -52,3 +51,6 @@ paddle-mobile.cbp
.idea .idea
cmake-build-debug/
exclude: 'third-party'
repos: repos:
- repo: https://github.com/Lucas-C/pre-commit-hooks.git - repo: https://github.com/Lucas-C/pre-commit-hooks.git
sha: v1.0.1 sha: v1.0.1
......
...@@ -50,14 +50,14 @@ int main() { ...@@ -50,14 +50,14 @@ int main() {
paddle_mobile::Loader<paddle_mobile::CPU> loader; paddle_mobile::Loader<paddle_mobile::CPU> loader;
auto program = loader.Load(std::string( auto program = loader.Load(std::string(
"../test/models/image_classification_resnet.inference.model")); "../../test/models/image_classification_resnet.inference.model"));
paddle_mobile::framework::Executor<paddle_mobile::CPU> executor(program); paddle_mobile::framework::Executor<paddle_mobile::CPU> executor(program);
paddle_mobile::framework::Tensor input; paddle_mobile::framework::Tensor input;
SetupTensor<float>(&input, {1, 3, 32, 32}, static_cast<float>(0), SetupTensor<float>(&input, {1, 3, 32, 32}, static_cast<float>(0),
static_cast<float>(1)); static_cast<float>(1));
float* input_ptr = input.data<float>(); float *input_ptr = input.data<float>();
for (int i = 0; i < input.numel(); ++i) { for (int i = 0; i < input.numel(); ++i) {
// std::cout << input_ptr[i] << std::endl; // std::cout << input_ptr[i] << std::endl;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册