yao_yf
5 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 2
- 2
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("--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("--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("--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("--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("--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") | parser.add_argument("--sparse", type=int, default=0, help="Enable sparse or not") | ||||
| @@ -84,7 +84,7 @@ class WideDeepConfig(): | |||||
| self.ckpt_path = "./" | self.ckpt_path = "./" | ||||
| self.stra_ckpt = './checkpoints/strategy.ckpt' | self.stra_ckpt = './checkpoints/strategy.ckpt' | ||||
| self.host_device_mix = 0 | self.host_device_mix = 0 | ||||
| self.dataset_type = "tfrecord" | |||||
| self.dataset_type = "mindrecord" | |||||
| self.parameter_server = 0 | self.parameter_server = 0 | ||||
| self.field_slice = False | self.field_slice = False | ||||
| self.manual_shape = None | self.manual_shape = None | ||||