31 lines
1.0 KiB
Bash
31 lines
1.0 KiB
Bash
CONFIG=input/config.json
|
|
OUTPUT=/app/output/$(python3 read_json.py $CONFIG task_id)
|
|
python3 load_config.py $CONFIG
|
|
echo "Convert PTQ YAML Haved been Prepared"
|
|
|
|
|
|
|
|
|
|
######### Img Adaptor
|
|
cd $OUTPUT/Img_Adaptor
|
|
BEGIN_IMG_ADAPTOR_TIME=$(date +%s)
|
|
echo -e "\033[44;37m===== Start Compiling Img Adaptor =====\033[0m"
|
|
hb_compile --config $OUTPUT/img_adaptor.yaml
|
|
echo -e "\033[44;37m===== End Compiling Img Adaptor =====\033[0m"
|
|
END_IMG_ADAPTOR_TIME=$(date +%s)
|
|
IMG_ADAPTOR_TIME=$((END_IMG_ADAPTOR_TIME - BEGIN_IMG_ADAPTOR_TIME))
|
|
echo -e "\033[44;37m===== Img Adaptor Time =====\033[0m"
|
|
echo -e "\033[44;37m===== $IMG_ADAPTOR_TIME seconds =====\033[0m"
|
|
|
|
########## DiT
|
|
cd $OUTPUT/DiT_Policy
|
|
BEGIN_DIT_TIME=$(date +%s)
|
|
echo -e "\033[44;37m===== Start Compiling DiT =====\033[0m"
|
|
hb_compile --config $OUTPUT/dit.yaml
|
|
echo -e "\033[44;37m===== End Compiling DiT =====\033[0m"
|
|
END_DIT_TIME=$(date +%s)
|
|
DIT_TIME=$((END_DIT_TIME - BEGIN_DIT_TIME))
|
|
echo -e "\033[44;37m===== DiT Time =====\033[0m"
|
|
echo -e "\033[44;37m===== $DIT_TIME seconds =====\033[0m"
|
|
|