diff --git a/mindspore/lite/src/dequant.h b/mindspore/lite/src/dequant.h
index fac1a86767..a3955eeb3c 100644
--- a/mindspore/lite/src/dequant.h
+++ b/mindspore/lite/src/dequant.h
@@ -66,6 +66,7 @@ class DequantUtil {
       if (!channel_first) {
         if (input_tensor->shape().size() != 2) {
           MS_LOG(ERROR) << "unexpected shape size: " << input_tensor->shape().size();
+          free(dequant_datas);
           return nullptr;
         }
         channels = input_tensor->shape()[1];
diff --git a/mindspore/lite/src/train/train_populate_parameter_v0.cc b/mindspore/lite/src/train/train_populate_parameter_v0.cc
index 5e67e385f2..44b19b14e8 100644
--- a/mindspore/lite/src/train/train_populate_parameter_v0.cc
+++ b/mindspore/lite/src/train/train_populate_parameter_v0.cc
@@ -290,6 +290,7 @@ OpParameter *PopulatePoolingGradParameter(const void *primitive) {
       break;
     default:
       MS_LOG(ERROR) << "unknown pooling mode: " << poolingGrad_prim->poolingMode();
+      free(pooling_param);
       return nullptr;
   }
 
diff --git a/mindspore/lite/src/train/train_session.cc b/mindspore/lite/src/train/train_session.cc
index 34d5ef2bc6..8dd5fc563c 100644
--- a/mindspore/lite/src/train/train_session.cc
+++ b/mindspore/lite/src/train/train_session.cc
@@ -74,7 +74,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;
 }
 
diff --git a/mindspore/lite/src/train/transfer_session.cc b/mindspore/lite/src/train/transfer_session.cc
index 71fbe8c809..7be49ca0fe 100644
--- a/mindspore/lite/src/train/transfer_session.cc
+++ b/mindspore/lite/src/train/transfer_session.cc
@@ -55,7 +55,8 @@ TransferSession::TransferSession(const char *model_buf_backbone, size_t size_bac
 
 std::vector<tensor::MSTensor *> TransferSession::GetInputs() const { return combined_inputs_; }
 
-bool TransferSession::CompileFormatTransform(tensor::MSTensor *out, tensor::MSTensor *in, int *mask) {
+bool TransferSession::CompileFormatTransform(tensor::MSTensor *out, tensor::MSTensor *in, int *mask, size_t mask_len) {
+  MS_ASSERT(out->shape().size() == mask_len);
   for (std::size_t dim = 0; dim != out->shape().size(); ++dim) {
     if (in->shape().at(mask[dim]) != out->shape().at(dim)) {
       return false;
@@ -85,7 +86,7 @@ int TransferSession::CompileTransferGraph() {
         }
         if (match == false && input->shape().size() == 4) {
           int nchw2nhwc_mask[4] = {0, 3, 1, 2};
-          nchw2nhwc_ = CompileFormatTransform(output, input, nchw2nhwc_mask);
+          nchw2nhwc_ = CompileFormatTransform(output, input, nchw2nhwc_mask, 4);
           match = nchw2nhwc_;
         }
         if (true == match) {
diff --git a/mindspore/lite/src/train/transfer_session.h b/mindspore/lite/src/train/transfer_session.h
index 69a08c8c54..3ad8219a8e 100644
--- a/mindspore/lite/src/train/transfer_session.h
+++ b/mindspore/lite/src/train/transfer_session.h
@@ -71,7 +71,7 @@ class TransferSession : public lite::TrainSession {
   bool is_valid_ = false;
 
  private:
-  bool CompileFormatTransform(tensor::MSTensor *out, tensor::MSTensor *in, int *mask);
+  bool CompileFormatTransform(tensor::MSTensor *out, tensor::MSTensor *in, int *mask, size_t mask_len);
   bool nchw2nhwc_ = false;
 };
 }  // namespace lite