From 98921a8d28fb9eeec206367d8ca8598aaf3603ef Mon Sep 17 00:00:00 2001 From: Haim Moushkatel Date: Sun, 25 Apr 2021 16:50:26 +0300 Subject: [PATCH] fixed issue I3O1BA --- mindspore/lite/src/train/train_session.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/lite/src/train/train_session.cc b/mindspore/lite/src/train/train_session.cc index 143000d392..f04d17cbcb 100644 --- a/mindspore/lite/src/train/train_session.cc +++ b/mindspore/lite/src/train/train_session.cc @@ -238,6 +238,7 @@ void TrainSession::CompileEvalOutputs() { if (eval_output_node_map_.find(in_kernel->name()) == eval_output_node_map_.end()) { auto *ms_tensor = in_kernel->out_tensors().at(0); if (ms_tensor != nullptr) { + ms_tensor->set_init_ref_count(ms_tensor->init_ref_count() + 1); eval_output_node_map_[in_kernel->name()].emplace_back(ms_tensor); auto index = TSFindTensor(tensors_, ms_tensor); if (index != tensors_.size()) {