1From ed8da9d72857e7fa34a7a389d37e0c8d4e70a4ef Mon Sep 17 00:00:00 2001 2From: z00574805 <z00574805@notesmail.huawei.com/> 3Date: Mon, 22 May 2023 16:40:32 +0800 4Subject: [PATCH 4/5] auto-apply 0001-fix-train-bug.patch 5 6--- 7 mindspore/lite/src/train/train_session.cc | 4 ++-- 8 1 file changed, 2 insertions(+), 2 deletions(-) 9 10diff --git a/mindspore/lite/src/train/train_session.cc b/mindspore/lite/src/train/train_session.cc 11index 2f9aa99b..e8dc2474 100644 12--- a/mindspore/lite/src/train/train_session.cc 13+++ b/mindspore/lite/src/train/train_session.cc 14@@ -222,8 +222,8 @@ static int ReshapeWeightTensor(Tensor *orig_tensor, lite::Tensor *new_tensor) { 15 16 int TrainSession::UpdateWeights(std::vector<lite::Tensor *> modify_tensors) { 17 unsigned int num_of_found_tensors = 0; 18- for (auto tensor : tensors_) { 19- for (auto modify : modify_tensors) { 20+ for (auto modify : modify_tensors) { 21+ for (auto tensor : tensors_) { 22 if (modify == nullptr) { 23 MS_LOG(ERROR) << "Tensor is nullptr"; 24 return RET_PARAM_INVALID; 25-- 262.17.1 27 28