diff --git a/benchmark/benchmark.yaml b/benchmark/benchmark.yaml index 844b7761fe488098ffe20ad5b3161ab96d0db4ea..3b1b9f0c0c0394be182c36bb337388ff39b22d5e 100644 --- a/benchmark/benchmark.yaml +++ b/benchmark/benchmark.yaml @@ -1,5 +1,5 @@ StyleGANv2: - dataset_web: https://paddlegan.bj.bcebos.com/datasets/ffhq.tar + dataset_web: https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar config: configs/stylegan_v2_256_ffhq.yaml fp_item: fp32 bs_item: 8 @@ -15,7 +15,7 @@ FOMM: log_interval: 1 esrgan: - dataset_web: https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14.tar + dataset_web: https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14paddle.tar config: configs/esrgan_psnr_x4_div2k.yaml fp_item: fp32 bs_item: 32 diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh index 386785da2276dbffc1716b156c9647df87e0e363..0dcfac1d282e2b6f813c6ef79dd71ec8749aab63 100644 --- a/test_tipc/prepare.sh +++ b/test_tipc/prepare.sh @@ -48,8 +48,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then cd ./data/ && tar xf cyclegan_horse2zebra_lite.tar && cd ../ ;; StyleGANv2) rm -rf ./data/ffhq* - wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq.tar --no-check-certificate - cd ./data/ && tar xf ffhq.tar && cd ../ ;; + wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate + cd ./data/ && tar xf ffhq_256.tar && cd ../ ;; FOMM) rm -rf ./data/fom_lite* wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/fom_lite.tar --no-check-certificate --no-check-certificate @@ -60,8 +60,8 @@ if [ ${MODE} = "lite_train_lite_infer" ];then cd ./data/ && tar xf reds_lite.tar && cd ../ ;; esrgan) rm -rf ./data/DIV2K* - wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14.tar --no-check-certificate - cd ./data/ && tar xf DIV2KandSet14.tar && cd ../ ;; + wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/DIV2KandSet14paddle.tar --no-check-certificate + cd ./data/ && tar xf DIV2KandSet14paddle.tar && cd ../ ;; singan) rm -rf ./data/singan* wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/singan-official_images.zip --no-check-certificate @@ -100,8 +100,8 @@ elif [ ${MODE} = "lite_train_whole_infer" ];then cd ./data/ && tar xf fom_lite.tar && cd ../ elif [ ${model_name} == "StyleGANv2" ]; then rm -rf ./data/ffhq* - wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq.tar --no-check-certificate - cd ./data/ && tar xf ffhq.tar && cd ../ + wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate + cd ./data/ && tar xf ffhq_256.tar && cd ../ elif [ ${model_name} == "basicvsr" ]; then rm -rf ./data/reds* wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/reds_lite.tar --no-check-certificate @@ -141,10 +141,10 @@ elif [ ${MODE} = "whole_infer" ];then cd ./inference && tar xf fom_dy2st.tar && cd ../ elif [ ${model_name} == "StyleGANv2" ]; then rm -rf ./data/ffhq* - wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq.tar --no-check-certificate + wget -nc -P ./data/ https://paddlegan.bj.bcebos.com/datasets/ffhq_256.tar --no-check-certificate wget -nc -P ./inference https://paddlegan.bj.bcebos.com/static_model/stylegan2_1024.tar --no-check-certificate cd ./inference && tar xf stylegan2_1024.tar && cd ../ - cd ./data/ && tar xf ffhq.tar && cd ../ + cd ./data/ && tar xf ffhq_256.tar && cd ../ elif [ ${model_name} == "basicvsr" ]; then rm -rf ./data/reds* rm -rf ./inference/basic*