diff --git a/mindspore/lite/src/train/train_session.cc b/mindspore/lite/src/train/train_session.cc
index def7e18fe9..99e11d96f4 100644
--- a/mindspore/lite/src/train/train_session.cc
+++ b/mindspore/lite/src/train/train_session.cc
@@ -73,7 +73,9 @@ std::unique_ptr<char[]> ReadFileToBuf(const std::string &filename, size_t *size)
     return std::unique_ptr<char[]>(nullptr);
   }
   ifs.close();
-  if (size) *size = fsize;
+  if (size != nullptr) {
+    *size = fsize;
+  }
   return buf;
 }