diff --git a/mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/TrainSession.java b/mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/TrainSession.java
index a341c07fc2..d76ec714e3 100644
--- a/mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/TrainSession.java
+++ b/mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/TrainSession.java
@@ -64,7 +64,7 @@ public class TrainSession {
 
     public MSTensor getInputsByTensorName(String tensorName) {
         Long tensorAddr = this.getInputsByTensorName(this.sessionPtr, tensorName);
-        if(tensorAddr == null) {
+        if (tensorAddr == null) {
             return null;
         }
         MSTensor msTensor = new MSTensor(tensorAddr);
@@ -99,7 +99,7 @@ public class TrainSession {
 
     public MSTensor getOutputByTensorName(String tensorName) {
         Long tensorAddr = getOutputByTensorName(this.sessionPtr, tensorName);
-        if(tensorAddr == null) {
+        if (tensorAddr == null) {
             return null;
         }
         return new MSTensor(tensorAddr);