()
| 241 | |
| 242 | |
| 243 | def parse_args(): |
| 244 | parser = argparse.ArgumentParser() |
| 245 | parser.add_argument( |
| 246 | "--test_predictor", |
| 247 | type=str, |
| 248 | default="False", |
| 249 | help="test paddle predictor") |
| 250 | parser.add_argument("--thread_num", type=int, default=2, help="thread num") |
| 251 | parser.add_argument("--batchsize", type=int, default=5, help="batch size") |
| 252 | parser.add_argument( |
| 253 | "--iteration_num", type=int, default=10, help="iteration num") |
| 254 | parser.add_argument( |
| 255 | "--reader_file", default="../data/out_test.1", type=str) |
| 256 | parser.add_argument( |
| 257 | "--params_file", default="../model/rec_inference.pdiparams", type=str) |
| 258 | parser.add_argument( |
| 259 | "--model_file", default="../model/rec_inference.pdmodel", type=str) |
| 260 | parser.add_argument("--log_file", default="./slot_dnn_infer.log", type=str) |
| 261 | parser.add_argument( |
| 262 | "--performance_file", default="./performance.txt", type=str) |
| 263 | parser.add_argument("--model_dir", type=str) |
| 264 | parser.add_argument("--use_gpu", type=str, default="False") |
| 265 | parser.add_argument("--data_dir", type=str) |
| 266 | parser.add_argument("--model_name", type=str, default="rec_model") |
| 267 | parser.add_argument("--cpu_threads", type=int, default=1) |
| 268 | parser.add_argument("--enable_mkldnn", type=str, default="False") |
| 269 | parser.add_argument("--enable_tensorRT", type=str, default="False") |
| 270 | args = parser.parse_args() |
| 271 | args.use_gpu = (True if args.use_gpu.lower() == "true" else False) |
| 272 | args.enable_mkldnn = (True |
| 273 | if args.enable_mkldnn.lower() == "true" else False) |
| 274 | args.enable_tensorRT = (True if args.enable_tensorRT.lower() == "true" else |
| 275 | False) |
| 276 | args.test_predictor = (True |
| 277 | if args.test_predictor.lower() == "true" else False) |
| 278 | return args |
| 279 | |
| 280 | |
| 281 | if __name__ == '__main__': |
no outgoing calls
no test coverage detected