From 7ee19dcfa05975d2af337c5af2416bbefa0d8d5b Mon Sep 17 00:00:00 2001 From: qiuleilei Date: Tue, 30 Sep 2025 17:31:51 +0800 Subject: [PATCH] bugfix --- docs/lite/docs/source_en/train/runtime_train_cpp.md | 9 +++++---- docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md | 9 +++++---- 2 files changed, 10 insertions(+), 8 deletions(-) diff --git a/docs/lite/docs/source_en/train/runtime_train_cpp.md b/docs/lite/docs/source_en/train/runtime_train_cpp.md index 024c2a0308..8ca6ca31f3 100644 --- a/docs/lite/docs/source_en/train/runtime_train_cpp.md +++ b/docs/lite/docs/source_en/train/runtime_train_cpp.md @@ -65,7 +65,7 @@ int CreateSession() { std::cout << "Error " << status << " during build of model " << ms_file_; MS_ASSERT(status != mindspore::kSuccess); } - return; + return status; } ``` @@ -154,10 +154,11 @@ float Evaluate() { test_ds_ = test_ds_->Map({&typecast}, {"label"}); test_ds_ = test_ds_->Batch(batch_size_, true); - model_->Evaluate(test_ds_, {}); - std::cout << "Accuracy is " << acc_metrics_->Eval() << std::endl; + auto acc_metrics_ = model_->Evaluate(test_ds_, {}); + auto res = acc_metrics_->Eval(); + std::cout << "Accuracy is " << res << std::endl; - return 0.0; + return res; } ``` diff --git a/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md b/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md index 8c098f52df..d8143cbeb5 100644 --- a/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md +++ b/docs/lite/docs/source_zh_cn/train/runtime_train_cpp.md @@ -65,7 +65,7 @@ int CreateSession() { std::cout << "Error " << status << " during build of model " << ms_file_; MS_ASSERT(status != mindspore::kSuccess); } - return; + return status; } ``` @@ -154,10 +154,11 @@ float Evaluate() { test_ds_ = test_ds_->Map({&typecast}, {"label"}); test_ds_ = test_ds_->Batch(batch_size_, true); - model_->Evaluate(test_ds_, {}); - std::cout << "Accuracy is " << acc_metrics_->Eval() << std::endl; + auto acc_metrics_ = model_->Evaluate(test_ds_, {}); + auto res = acc_metrics_->Eval(); + std::cout << "Accuracy is " << res << std::endl; - return 0.0; + return res; } ``` -- Gitee