| 297 | |
| 298 | |
| 299 | def get_arg_parser(): |
| 300 | parser = argparse.ArgumentParser() |
| 301 | parser.add_argument('--data_location', |
| 302 | help='Full path of train data', |
| 303 | required=False, |
| 304 | default='./data') |
| 305 | parser.add_argument('--steps', |
| 306 | help='set the number of steps on train dataset', |
| 307 | type=int, |
| 308 | default=0) |
| 309 | parser.add_argument('--batch_size', |
| 310 | help='Batch size to train. Default is 512', |
| 311 | type=int, |
| 312 | default=512) |
| 313 | parser.add_argument('--output_dir', |
| 314 | help='Full path to logs & model output directory', |
| 315 | required=False, |
| 316 | default='./result') |
| 317 | parser.add_argument('--checkpoint', |
| 318 | help='Full path to checkpoints input/output directory', |
| 319 | required=False) |
| 320 | parser.add_argument('--deep_dropout', |
| 321 | help='Dropout regularization for deep model', |
| 322 | type=float, |
| 323 | default=0.0) |
| 324 | parser.add_argument('--learning_rate', |
| 325 | help='Learning rate for model', |
| 326 | type=float, |
| 327 | default=0.1) |
| 328 | parser.add_argument('--save_steps', |
| 329 | help='set the number of steps on saving checkpoints', |
| 330 | type=int, |
| 331 | default=0) |
| 332 | parser.add_argument('--keep_checkpoint_max', |
| 333 | help='Maximum number of recent checkpoint to keep', |
| 334 | type=int, |
| 335 | default=1) |
| 336 | parser.add_argument('--bf16', |
| 337 | help='enable DeepRec BF16 in deep model. Default FP32', |
| 338 | action='store_true') |
| 339 | parser.add_argument('--no_eval', |
| 340 | help='not evaluate trained model by eval dataset.', |
| 341 | action='store_true') |
| 342 | parser.add_argument('--timeline', |
| 343 | help='number of steps on saving timeline. Default 0', |
| 344 | type=int, |
| 345 | default=0) |
| 346 | parser.add_argument("--interaction_op", |
| 347 | type=str, |
| 348 | choices=["dot", "cat"], |
| 349 | default="cat") |
| 350 | parser.add_argument("--protocol", |
| 351 | type=str, |
| 352 | choices=["grpc", "grpc++", "star_server"], |
| 353 | default="grpc") |
| 354 | parser.add_argument('--inter', |
| 355 | help='set inter op parallelism threads.', |
| 356 | type=int, |