diff --git a/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py b/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py index f3944d07b..c5180ca10 100644 --- a/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py +++ b/ppcls/arch/backbone/model_zoo/swin_transformer_v2.py @@ -28,23 +28,23 @@ from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_fro MODEL_URLS = { "SwinTransformerV2_tiny_patch4_window8_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_tiny_patch4_window8_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_tiny_patch4_window8_256_pretrained.pdparams", "SwinTransformerV2_tiny_patch4_window16_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_tiny_patch4_window16_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_tiny_patch4_window16_256_pretrained.pdparams", "SwinTransformerV2_small_patch4_window8_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_small_patch4_window8_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_small_patch4_window8_256_pretrained.pdparams", "SwinTransformerV2_small_patch4_window16_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_small_patch4_window16_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_small_patch4_window16_256_pretrained.pdparams", "SwinTransformerV2_base_patch4_window8_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window8_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window8_256_pretrained.pdparams", "SwinTransformerV2_base_patch4_window16_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window16_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window16_256_pretrained.pdparams", "SwinTransformerV2_base_patch4_window24_384": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window24_384_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window24_384_pretrained.pdparams", "SwinTransformerV2_large_patch4_window16_256": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_large_patch4_window16_256_pretrained.pdparams", + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_large_patch4_window16_256_pretrained.pdparams", "SwinTransformerV2_large_patch4_window24_384": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_large_patch4_window24_384_pretrained.pdparams" + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_large_patch4_window24_384_pretrained.pdparams" } __all__ = list(MODEL_URLS.keys()) @@ -380,7 +380,7 @@ class SwinTransformerBlock(nn.Layer): assert L == H * W, "input feature has wrong size" shortcut = x - x = x.reshape(shape=[B, H, W, C]) + x = x.reshape([B, H, W, C]) # cyclic shift if self.shift_size > 0: @@ -782,8 +782,8 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=use_ssld, - use_imagenet22k_pretrained=False, - use_imagenet22kto1k_pretrained=False) + use_imagenet22k_pretrained=use_imagenet22k_pretrained, + use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained) elif isinstance(pretrained, str): load_dygraph_pretrain(model, pretrained, **kwargs) else: @@ -916,7 +916,7 @@ def SwinTransformerV2_base_patch4_window24_384( pretrained=False, use_ssld=False, use_imagenet22k_pretrained=False, - use_imagenet22kto1k_pretrained=False, + use_imagenet22kto1k_pretrained=True, **kwargs): model = SwinTransformerV2( img_size=384, @@ -941,7 +941,7 @@ def SwinTransformerV2_large_patch4_window16_256( pretrained=False, use_ssld=False, use_imagenet22k_pretrained=False, - use_imagenet22kto1k_pretrained=False, + use_imagenet22kto1k_pretrained=True, **kwargs): model = SwinTransformerV2( img_size=256, @@ -966,7 +966,7 @@ def SwinTransformerV2_large_patch4_window24_384( pretrained=False, use_ssld=False, use_imagenet22k_pretrained=False, - use_imagenet22kto1k_pretrained=False, + use_imagenet22kto1k_pretrained=True, **kwargs): model = SwinTransformerV2( img_size=384, diff --git a/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384.yaml b/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384.yaml index 123c5f6b2..95372158a 100644 --- a/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384.yaml @@ -123,7 +123,7 @@ DataLoader: order: '' sampler: name: DistributedBatchSampler - batch_size: 32 + batch_size: 128 drop_last: False shuffle: False loader: diff --git a/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384.yaml b/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384.yaml index 87d30a15a..0ed9414e3 100644 --- a/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384.yaml @@ -94,7 +94,7 @@ DataLoader: sampler: name: DistributedBatchSampler - batch_size: 16 + batch_size: 128 drop_last: False shuffle: True loader: @@ -123,7 +123,7 @@ DataLoader: order: '' sampler: name: DistributedBatchSampler - batch_size: 16 + batch_size: 128 drop_last: False shuffle: False loader: diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window16_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window16_256_train_infer_python.txt index 6588cdee7..fa79ef1e7 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window16_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window16_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384_train_infer_python.txt index affc0658b..20f4a65c9 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=384 -o PreProcess.transform_ops.1.CropImage.size=384 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window8_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window8_256_train_infer_python.txt index 489ee8c2c..0778efe01 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window8_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window8_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window16_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window16_256_train_infer_python.txt index e31ee46e1..b02614499 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window16_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window16_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384_train_infer_python.txt index b2a8b5e43..64a7fa7a8 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=384 -o PreProcess.transform_ops.1.CropImage.size=384 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window16_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window16_256_train_infer_python.txt index ba4352b80..ed6a9a311 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window16_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window16_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window8_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window8_256_train_infer_python.txt index 50fb82018..1e6475092 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window8_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window8_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window16_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window16_256_train_infer_python.txt index 81273d2e1..3eee30a8b 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window16_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window16_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1 diff --git a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window8_256_train_infer_python.txt b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window8_256_train_infer_python.txt index b7d8858d4..615102aae 100644 --- a/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window8_256_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window8_256_train_infer_python.txt @@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S 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 +inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256 +-o Global.use_gpu:True -o Global.enable_mkldnn:False -o Global.cpu_num_threads:1 -o Global.batch_size:1