Browse Source

fixed issue I3O1BA

pull/15661/head
Haim Moushkatel 4 years ago
parent
commit
98921a8d28
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/train/train_session.cc

+ 1
- 0
mindspore/lite/src/train/train_session.cc View File

@@ -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()) {


Loading…
Cancel
Save