apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Backend] fix lite backend save model error (#474)

Browse Source 
* [Backend] fix lite backend save model error

* [Backend] fixed typos
This commit is contained in:
DefTruth
2022-11-02 10:01:02 +08:00
committed by GitHub
parent e5c955dd3e
commit d8aa11efc8
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
fastdeploy/backends/lite/lite_backend.cc
View File

@@ -45,6 +45,8 @@ void LiteBackend::BuildOption(const LiteBackendOption& option) {
if (option_.enable_int8) { if (option_.enable_int8) {
valid_places.push_back( valid_places.push_back(
paddle::lite_api::Place{TARGET(kARM), PRECISION(kInt8)}); paddle::lite_api::Place{TARGET(kARM), PRECISION(kInt8)});
FDINFO << "Lite::Backend enable_int8 option is ON ! Lite::Backend will "
<< "inference with int8 precision!" << std::endl;
} }
if (option_.enable_fp16) { if (option_.enable_fp16) {
paddle::lite_api::MobileConfig check_fp16_config; paddle::lite_api::MobileConfig check_fp16_config;
@@ -55,6 +57,8 @@ void LiteBackend::BuildOption(const LiteBackendOption& option) {
if (supported_fp16_) { if (supported_fp16_) {
valid_places.push_back( valid_places.push_back(
paddle::lite_api::Place{TARGET(kARM), PRECISION(kFP16)}); paddle::lite_api::Place{TARGET(kARM), PRECISION(kFP16)});
FDINFO << "Your device is supported fp16 ! Lite::Backend will "
<< "inference with fp16 precision!" << std::endl;
} else { } else {
FDWARNING << "This device is not supported fp16, will skip fp16 option."; FDWARNING << "This device is not supported fp16, will skip fp16 option.";
} }
@@ -89,7 +93,8 @@ bool LiteBackend::InitFromPaddle(const std::string& model_file,
if (option_.optimized_model_dir != "") { if (option_.optimized_model_dir != "") {
FDINFO << "Optimzed model dir is not empty, will save optimized model to: " FDINFO << "Optimzed model dir is not empty, will save optimized model to: "
<< option_.optimized_model_dir << std::endl; << option_.optimized_model_dir << std::endl;
predictor_->SaveOptimizedModel(option_.optimized_model_dir); predictor_->SaveOptimizedModel(option_.optimized_model_dir,
paddle::lite_api::LiteModelType::kNaiveBuffer);
} }
inputs_desc_.clear(); inputs_desc_.clear();