Skip to content

Commit

Permalink
add tipc (PaddlePaddle#492)
Browse files Browse the repository at this point in the history
  • Loading branch information
lzzyzlbb authored Nov 23, 2021
1 parent 2ab96cb commit 3d4fa14
Show file tree
Hide file tree
Showing 10 changed files with 946 additions and 0 deletions.
65 changes: 65 additions & 0 deletions test_tipc/common_func.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
#!/bin/bash

function func_parser_key(){
strs=$1
IFS=":"
array=(${strs})
tmp=${array[0]}
echo ${tmp}
}

function func_parser_value(){
strs=$1
IFS=":"
array=(${strs})
tmp=${array[1]}
echo ${tmp}
}

function func_set_params(){
key=$1
value=$2
if [ ${key}x = "null"x ];then
echo " "
elif [[ ${value} = "null" ]] || [[ ${value} = " " ]] || [ ${#value} -le 0 ];then
echo " "
else
echo "${key}=${value}"
fi
}

function func_parser_params(){
strs=$1
IFS=":"
array=(${strs})
key=${array[0]}
tmp=${array[1]}
IFS="|"
res=""
for _params in ${tmp[*]}; do
IFS="="
array=(${_params})
mode=${array[0]}
value=${array[1]}
if [[ ${mode} = ${MODE} ]]; then
IFS="|"
#echo $(func_set_params "${mode}" "${value}")
echo $value
break
fi
IFS="|"
done
echo ${res}
}

function status_check(){
last_status=$1 # the exit code
run_command=$2
run_log=$3
if [ $last_status -eq 0 ]; then
echo -e "\033[33m Run successfully with command - ${run_command}! \033[0m" | tee -a ${run_log}
else
echo -e "\033[33m Run failed with command - ${run_command}! \033[0m" | tee -a ${run_log}
fi
}

51 changes: 51 additions & 0 deletions test_tipc/configs/basicvsr/train_infer_python.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
===========================train_params===========================
model_name:basicvsr
python:python3.7
gpu_list:0
##
auto_cast:null
total_iters:lite_train_lite_infer=5|whole_train_whole_infer=200
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=1|whole_train_whole_infer=1
pretrained_model:null
train_model_name:basicvsr_reds*/*checkpoint.pdparams
train_infer_img_dir:./data/basicvsr_reds/test
null:null
##
trainer:norm_train
norm_train:tools/main.py -c configs/basicvsr_reds.yaml -o dataset.train.dataset.num_clips=2
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
null:null
##
===========================infer_params===========================
--output_dir:./output/
load:null
norm_export:tools/export_model.py -c configs/basicvsr_reds.yaml --inputs_size="1,6,3,180,320" --load
quant_export:null
fpgm_export:null
distill_export:null
export1:null
export2:null
inference_dir:basicvsrmodel_generator
train_model:./inference/basicvsr/basicvsrmodel_generator
infer_export:null
infer_quant:False
inference:tools/inference.py --model_type basicvsr -c configs/basicvsr_reds.yaml -o dataset.test.num_clips=2 dataset.test.number_frames=6
--device:gpu
null:null
null:null
null:null
null:null
null:null
--model_path:
null:null
null:null
--benchmark:True
null:null
51 changes: 51 additions & 0 deletions test_tipc/configs/cyclegan/train_infer_python.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
===========================train_params===========================
model_name:cyclegan
python:python3.7
gpu_list:0|0,1
##
auto_cast:null
epochs:lite_train_lite_infer=5|whole_train_whole_infer=200
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=1|whole_train_whole_infer=1
pretrained_model:null
train_model_name:cyclegan_horse2zebra*/*checkpoint.pdparams
train_infer_img_dir:./data/horse2zebra/test
null:null
##
trainer:norm_train
norm_train:tools/main.py -c configs/cyclegan_horse2zebra.yaml -o
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
null:null
##
===========================infer_params===========================
--output_dir:./output/
load:null
norm_export:tools/export_model.py -c configs/cyclegan_horse2zebra.yaml --inputs_size="-1,3,-1,-1;-1,3,-1,-1" --load
quant_export:null
fpgm_export:null
distill_export:null
export1:null
export2:null
inference_dir:cycleganmodel_netG_A
train_model:./inference/cyclegan_horse2zebra/cycleganmodel_netG_A
infer_export:null
infer_quant:False
inference:tools/inference.py --model_type cyclegan -c configs/cyclegan_horse2zebra.yaml
--device:gpu
null:null
null:null
null:null
null:null
null:null
--model_path:
null:null
null:null
--benchmark:True
null:null
51 changes: 51 additions & 0 deletions test_tipc/configs/fom/train_infer_python.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
===========================train_params===========================
model_name:fom
python:python3.7
gpu_list:0
##
auto_cast:null
epochs:lite_train_lite_infer=10|whole_train_whole_infer=100
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=8|whole_train_whole_infer=8
pretrained_model:null
train_model_name:firstorder_vox_256*/*checkpoint.pdparams
train_infer_img_dir:./data/firstorder_vox_256/test
null:null
##
trainer:norm_train
norm_train:tools/main.py -c configs/firstorder_vox_256.yaml -o
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
null:null
##
===========================infer_params===========================
--output_dir:./output/
load:null
norm_export:tools/export_model.py -c configs/firstorder_vox_256.yaml --inputs_size="1,3,256,256;1,3,256,256;1,10,2;1,10,2,2" --load
quant_export:null
fpgm_export:null
distill_export:null
export1:null
export2:null
inference_dir:fom_dy2st
train_model:./inference/fom_dy2st/
infer_export:null
infer_quant:False
inference:tools/fom_infer.py --driving_path data/first_order/Voxceleb/test --output_path infer_output/fom
--device:gpu
null:null
null:null
null:null
null:null
null:null
--model_path:
null:null
null:null
--benchmark:True
null:null
51 changes: 51 additions & 0 deletions test_tipc/configs/pix2pix/train_infer_python.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
===========================train_params===========================
model_name:pix2pix
python:python3.7
gpu_list:0|0,1
##
auto_cast:null
epochs:lite_train_lite_infer=5|whole_train_whole_infer=200
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=1|whole_train_whole_infer=1
pretrained_model:null
train_model_name:pix2pix_facades*/*checkpoint.pdparams
train_infer_img_dir:./data/facades/test
null:null
##
trainer:norm_train
norm_train:tools/main.py -c configs/pix2pix_facades.yaml -o
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
null:null
##
===========================infer_params===========================
--output_dir:./output/
load:null
norm_export:tools/export_model.py -c configs/pix2pix_facades.yaml --inputs_size="-1,3,-1,-1" --load
quant_export:null
fpgm_export:null
distill_export:null
export1:null
export2:null
inference_dir:pix2pixmodel_netG
train_model:./inference/pix2pix_facade/pix2pixmodel_netG
infer_export:null
infer_quant:False
inference:tools/inference.py --model_type pix2pix -c configs/pix2pix_facades.yaml
--device:cpu
null:null
null:null
null:null
null:null
null:null
--model_path:
null:null
null:null
--benchmark:True
null:null
51 changes: 51 additions & 0 deletions test_tipc/configs/stylegan2/train_infer_python.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
===========================train_params===========================
model_name:stylegan2
python:python3.7
gpu_list:0
##
auto_cast:null
total_iters::lite_train_lite_infer=10|whole_train_whole_infer=800
output_dir:./output/
dataset.train.batch_size:lite_train_lite_infer=3|whole_train_whole_infer=3
pretrained_model:null
train_model_name:stylegan_v2_256_ffhq*/*checkpoint.pdparams
train_infer_img_dir:null
null:null
##
trainer:norm_train
norm_train:tools/main.py -c configs/stylegan_v2_256_ffhq.yaml -o
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
null:null
##
===========================infer_params===========================
--output_dir:./output/
load:null
norm_export:tools/export_model.py -c configs/stylegan_v2_256_ffhq.yaml --inputs_size="1,1,512;1,1" --load
quant_export:null
fpgm_export:null
distill_export:null
export1:null
export2:null
inference_dir:stylegan2model_gen
train_model:./inference/stylegan2/stylegan2model_gen
infer_export:null
infer_quant:False
inference:tools/inference.py --model_type stylegan2 -c configs/stylegan_v2_256_ffhq.yaml
--device:gpu
null:null
null:null
null:null
null:null
null:null
--model_path:
null:null
null:null
--benchmark:True
null:null
Loading

0 comments on commit 3d4fa14

Please sign in to comment.