diff --git a/src/io/loader.cpp b/src/io/loader.cpp index 7a0912106db8acaafb751d2c467245ee302c84e6..48a2b5cfdaa5f53cd9611dd0be1ce3df05988311 100644 --- a/src/io/loader.cpp +++ b/src/io/loader.cpp @@ -33,9 +33,7 @@ void InitMemoryFromProgram( for (const auto &var_desc : block->Vars()) { auto var = scope.get()->Var(var_desc->Name()); if (var_desc->Type() == framework::VARTYPE_TYPE_LOD_TENSOR) { - if (var_desc->Persistable() && - var_desc->Type() != framework::VARTYPE_TYPE_FEED_MINIBATCH && - var_desc->Type() != framework::VARTYPE_TYPE_FETCH_LIST) { + if (var_desc->Persistable()) { auto dim = var_desc->Tensor_desc().Dims(); auto tensor = var->GetMutable(); tensor->Resize(framework::make_ddim(dim));