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