refine save hook (#48124)

04709310 · wanghuancoder · GitHub · d595928e · 04709310
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/eager/tensor_wrapper.h paddle/fluid/eager/tensor_wrapper.h +1 -1

未找到文件。
--- a/paddle/fluid/eager/tensor_wrapper.h
+++ b/paddle/fluid/eager/tensor_wrapper.h
@@ -74,7 +74,7 @@ class TensorWrapper {
    } else {
 #ifndef PADDLE_NO_PYTHON
      if (SavedTensorsHooks::GetInstance().IsEnable() &&
-          tensor.is_dense_tensor()) {
+          tensor.is_dense_tensor() && tensor.initialized()) {
        phi::DenseTensor* dense_tensor =
            static_cast<phi::DenseTensor*>(tensor.impl().get());
        intermidiate_tensor_.set_impl(