From 48e6c9aec5bb73f231d79f2c690a29c9f8bc9376 Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Tue, 4 Sep 2018 11:42:34 +0800 Subject: [PATCH] small clean --- src/io/loader.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/io/loader.cpp b/src/io/loader.cpp index 7a0912106d..48a2b5cfda 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)); -- GitLab