From 83637b1d1d5f53460d5dd16d8ec26000a46d1f7f Mon Sep 17 00:00:00 2001 From: zhangyubo0722 Date: Tue, 14 Mar 2023 08:20:19 +0000 Subject: [PATCH] add_tipc_config_7 --- .../AlexNet/AlexNet_train_infer_python.txt | 11 +++- ...atch16_224_finetune_train_infer_python.txt | 61 +++++++++++++++++++ ...iT_base_patch16_224_train_infer_python.txt | 11 +++- 3 files changed, 79 insertions(+), 4 deletions(-) create mode 100644 test_tipc/configs/CAE/cae_base_patch16_224_finetune_train_infer_python.txt diff --git a/test_tipc/configs/AlexNet/AlexNet_train_infer_python.txt b/test_tipc/configs/AlexNet/AlexNet_train_infer_python.txt index 990b67f8..9a05a0e3 100644 --- a/test_tipc/configs/AlexNet/AlexNet_train_infer_python.txt +++ b/test_tipc/configs/AlexNet/AlexNet_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/AlexNet/AlexNet.yaml -o Glob pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile +flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/CAE/cae_base_patch16_224_finetune_train_infer_python.txt b/test_tipc/configs/CAE/cae_base_patch16_224_finetune_train_infer_python.txt new file mode 100644 index 00000000..26045f1e --- /dev/null +++ b/test_tipc/configs/CAE/cae_base_patch16_224_finetune_train_infer_python.txt @@ -0,0 +1,61 @@ +===========================train_params=========================== +model_name:cae_base_patch16_224 +python:python3.7 +gpu_list:0|0,1 +-o Global.device:gpu +-o Global.auto_cast:null +-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120 +-o Global.output_dir:./output/ +-o DataLoader.Train.sampler.batch_size:8 +-o Global.pretrained_model:null +train_model_name:latest +train_infer_img_dir:./dataset/ILSVRC2012/val +null:null +## +trainer:norm_train +norm_train:tools/train.py -c ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml -o Global.seed=1234 -o DataLoader.Train.sampler.shuffle=False -o DataLoader.Train.loader.num_workers=0 -o DataLoader.Train.loader.use_shared_memory=False -o Global.eval_during_train=False -o Global.save_interval=2 +pact_train:null +fpgm_train:null +distill_train:null +to_static_train:-o Global.to_static=True +null:null +## +===========================eval_params=========================== +eval:tools/eval.py -c ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml +null:null +## +===========================infer_params========================== +-o Global.save_inference_dir:./inference +-o Global.pretrained_model: +norm_export:tools/export_model.py -c ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml +quant_export:null +fpgm_export:null +distill_export:null +kl_quant:null +export2:null +pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/cae_base_patch16_224_pretrained.pdparams +infer_model:../inference/ +infer_export:True +infer_quant:Fasle +inference:python/predict_cls.py -c configs/inference_cls.yaml +-o Global.use_gpu:True|False +-o Global.enable_mkldnn:False +-o Global.cpu_num_threads:1 +-o Global.batch_size:1 +-o Global.use_tensorrt:False +-o Global.use_fp16:False +-o Global.inference_model_dir:../inference +-o Global.infer_imgs:../dataset/ILSVRC2012/val/ILSVRC2012_val_00000001.JPEG +-o Global.save_log_path:null +-o Global.benchmark:False +null:null +null:null +===========================train_benchmark_params========================== +batch_size:16 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile +flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/VisionTransformer/ViT_base_patch16_224_train_infer_python.txt b/test_tipc/configs/VisionTransformer/ViT_base_patch16_224_train_infer_python.txt index 7edde6d2..3e208dd9 100644 --- a/test_tipc/configs/VisionTransformer/ViT_base_patch16_224_train_infer_python.txt +++ b/test_tipc/configs/VisionTransformer/ViT_base_patch16_224_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/VisionTransformer/ViT_base_p pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:64 +fp_items:fp32|fp16 +epoch:1 +model_type:norm_train|to_static_train +--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile +flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] -- GitLab