diff --git a/build.sh b/build.sh
index f0dc6956..b693ba74 100644
--- a/build.sh
+++ b/build.sh
@@ -59,7 +59,7 @@ checkopts()
   ENABLE_GE_ST="off"
   ENABLE_GE_COV="off"
   GE_ONLY="on"
-  PLATFORM="inference"
+  PLATFORM="train"
   PRODUCT="normal"
   ENABLE_GITEE="off"
   # Process the options