58 lines
2.1 KiB
Plaintext
58 lines
2.1 KiB
Plaintext
===========================train_params===========================
|
|
model_name:ResNet50_vd|ResNeXt101_vd_64x4d|HRNet_W18_C|MobileNetV3_large_x1_0|DarkNet53|MobileNetV1|MobileNetV2|ShuffleNetV2_x1_0
|
|
model_name_pact:ResNet50_vd|MobileNetV3_large_x1_0
|
|
model_name_fpgm:ResNet50_vd|MobileNetV3_large_x1_0
|
|
model_name_kl:ResNet50_vd|MobileNetV3_large_x1_0
|
|
python:python3.7
|
|
gpu_list:0|0,1|-1
|
|
Global.epoch_num:10
|
|
Global.save_model_dir:./output/
|
|
Global.pretrained_model:null
|
|
Global.save_inference_dir:null
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
===========================scripts===========================
|
|
train:tools/train.py -c test_ci/benchmark.yaml
|
|
eval:tools/eval.py -c test_ci/benchmark.yaml
|
|
norm_export:tools/export_model.py -c test_ci/benchmark.yaml
|
|
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
===========================infer_params===========================
|
|
Global.save_log_path:./test/output/
|
|
Global.use_gpu:True|False
|
|
Global.enable_mkldnn:True|False
|
|
Global.cpu_num_threads:1|6
|
|
Global.batch_size:1
|
|
Global.use_tensorrt:True|False
|
|
Global.use_fp16:True|False
|
|
Global.inference_model_dir:./inference
|
|
Global.infer_imgs:./dataset/chain_dataset/val
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
===========================pretrained_model===========================
|
|
ResNet50_vd:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/ResNet50_vd_pretrained.pdparams
|
|
ResNeXt101_vd_64x4d:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ResNeXt101_vd_64x4d_pretrained.pdparams
|
|
HRNet_W18_C:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/HRNet_W18_C_pretrained.pdparams
|
|
MobileNetV3_large_x1_0:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/MobileNetV3_large_x1_0_pretrained.pdparams
|
|
DarkNet53:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/DarkNet53_pretrained.pdparams
|
|
MobileNetV1:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/MobileNetV1_pretrained.pdparams
|
|
MobileNetV2:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV2_pretrained.pdparams
|
|
ShuffleNetV2_x1_0:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/ShuffleNetV2_x1_0_pretrained.pdparams
|