support to_static

pull/2648/head
gaotingquan 2023-01-31 09:28:24 +00:00 committed by Tingquan Gao
parent ada7a3becf
commit 2fc143f338
1 changed files with 3 additions and 3 deletions

View File

@ -17,10 +17,10 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/DenseNet/DenseNet121.yaml -o
pact_train:null
fpgm_train:null
distill_train:null
null:null
to_static_train:-o Global.to_static=True
null:null
##
===========================eval_params===========================
===========================eval_params===========================
eval:tools/eval.py -c ppcls/configs/ImageNet/DenseNet/DenseNet121.yaml
null:null
##
@ -58,4 +58,4 @@ 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]}]
random_infer_input:[{float32,[3,224,224]}]