mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

modify tipc

Browse Source
This commit is contained in:
zhangyubo0722 2023-01-12 07:50:50 +00:00 committed by cuicheng01
parent d7e071fb8d
commit add3ea238c
12 changed files with 36 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
ppcls/arch/backbone/model_zoo/swin_transformer_v2.py
View File

@ -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,

2
ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384.yaml
View File

@ -123,7 +123,7 @@ DataLoader:
order: ''
sampler:
name: DistributedBatchSampler
batch_size: 32
batch_size: 128
drop_last: False
shuffle: False
loader:

4
ppcls/configs/ImageNet/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384.yaml
View File

@ -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:

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window16_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window24_384_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_base_patch4_window8_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window16_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_large_patch4_window24_384_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window16_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_small_patch4_window8_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window16_256_train_infer_python.txt
View File

@ -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

4
test_tipc/configs/SwinTransformerV2/SwinTransformerV2_tiny_patch4_window8_256_train_infer_python.txt
View File

@ -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