diff --git a/model_zoo/official/recommend/wide_and_deep/src/config.py b/model_zoo/official/recommend/wide_and_deep/src/config.py
index aeafdc39f7..1c206854b6 100644
--- a/model_zoo/official/recommend/wide_and_deep/src/config.py
+++ b/model_zoo/official/recommend/wide_and_deep/src/config.py
@@ -45,7 +45,7 @@ def argparse_init():
     parser.add_argument("--eval_file_name", type=str, default="eval.log", help="Eval output file.")
     parser.add_argument("--loss_file_name", type=str, default="loss.log", help="Loss output file.")
     parser.add_argument("--host_device_mix", type=int, default=0, help="Enable host device mode or not")
-    parser.add_argument("--dataset_type", type=str, default="tfrecord", help="tfrecord/mindrecord/hd5")
+    parser.add_argument("--dataset_type", type=str, default="mindrecord", help="tfrecord/mindrecord/hd5")
     parser.add_argument("--parameter_server", type=int, default=0, help="Open parameter server of not")
     parser.add_argument("--field_slice", type=int, default=0, help="Enable split field mode or not")
     parser.add_argument("--sparse", type=int, default=0, help="Enable sparse or not")
@@ -84,7 +84,7 @@ class WideDeepConfig():
         self.ckpt_path = "./"
         self.stra_ckpt = './checkpoints/strategy.ckpt'
         self.host_device_mix = 0
-        self.dataset_type = "tfrecord"
+        self.dataset_type = "mindrecord"
         self.parameter_server = 0
         self.field_slice = False
         self.manual_shape = None