浏览代码

change wide_and_deep default dataset type to mindrecord

tags/v1.1.0
yao_yf 5 年前
父节点
当前提交
30b97f455e
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. +2
    -2
      model_zoo/official/recommend/wide_and_deep/src/config.py

+ 2
- 2
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


正在加载...
取消
保存