p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!10075 change wide&deep default dataset type to mindrecord 

From: @yao_yf
Reviewed-by: @stsuteng,@kisnwang
Signed-off-by: @stsuteng
This commit is contained in:
mindspore-ci-bot 2020-12-17 11:22:09 +08:00 committed by Gitee
parent 647b697cf9 30b97f455e
commit 153676fdd0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/official/recommend/wide_and_deep/src/config.py
View File

@ -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