You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
75 lines
3.4 KiB
75 lines
3.4 KiB
2 years ago
|
|
||
|
|
|
||
|
|
def configs(model_name):
|
||
|
|
args = {
|
||
|
|
'swin_base_patch244_window877_kinetics400_1k':
|
||
|
|
{'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_base_patch244_window877_kinetics400_1k.pth',
|
||
|
|
'num_classes': 400,
|
||
|
|
'labels_file_name': 'kinetics_400.json',
|
||
|
|
'embed_dim': 128,
|
||
|
|
'depths': [2, 2, 18, 2],
|
||
|
|
'num_heads': [4, 8, 16, 32],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (8, 7, 7), 'drop_path_rate': 0.4, 'patch_norm': True},
|
||
|
|
'swin_small_patch244_window877_kinetics400_1k':
|
||
|
|
{
|
||
|
|
'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_small_patch244_window877_kinetics400_1k.pth',
|
||
|
|
'num_classes': 400,
|
||
|
|
'labels_file_name': 'kinetics_400.json',
|
||
|
|
'embed_dim': 96,
|
||
|
|
'depths': [2, 2, 18, 2],
|
||
|
|
'num_heads': [3, 6, 12, 24],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (8, 7, 7),
|
||
|
|
'drop_path_rate': 0.4,
|
||
|
|
'patch_norm': True},
|
||
|
|
'swin_tiny_patch244_window877_kinetics400_1k':
|
||
|
|
{
|
||
|
|
'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_tiny_patch244_window877_kinetics400_1k.pth',
|
||
|
|
'num_classes': 400,
|
||
|
|
'labels_file_name': 'kinetics_400.json',
|
||
|
|
'embed_dim': 96,
|
||
|
|
'depths': [2, 2, 6, 2],
|
||
|
|
'num_heads': [3, 6, 12, 24],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (8, 7, 7),
|
||
|
|
'drop_path_rate': 0.1,
|
||
|
|
'patch_norm': True},
|
||
|
|
'swin_base_patch244_window877_kinetics400_22k':
|
||
|
|
{
|
||
|
|
'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_base_patch244_window877_kinetics400_22k.pth',
|
||
|
|
'num_classes': 400,
|
||
|
|
'labels_file_name': 'kinetics_400.json',
|
||
|
|
'embed_dim': 128,
|
||
|
|
'depths': [2, 2, 18, 2],
|
||
|
|
'num_heads': [4, 8, 16, 32],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (8, 7, 7),
|
||
|
|
'drop_path_rate': 0.4,
|
||
|
|
'patch_norm': True},
|
||
|
|
'swin_base_patch244_window877_kinetics600_22k':
|
||
|
|
{
|
||
|
|
'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_base_patch244_window877_kinetics600_22k.pth',
|
||
|
|
'num_classes': 600,
|
||
|
|
'labels_file_name': '',
|
||
|
|
'embed_dim': 128,
|
||
|
|
'depths': [2, 2, 18, 2],
|
||
|
|
'num_heads': [4, 8, 16, 32],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (8, 7, 7), 'drop_path_rate': 0.4, 'patch_norm': True},
|
||
|
|
'swin_base_patch244_window1677_sthv2':
|
||
|
|
{
|
||
|
|
'pretrained': 'https://github.com/SwinTransformer/storage/releases/download/v1.0.4/swin_base_patch244_window1677_sthv2.pth',
|
||
|
|
'num_classes': 174,
|
||
|
|
'labels_file_name': '',
|
||
|
|
'embed_dim': 128,
|
||
|
|
'depths': [2, 2, 18, 2],
|
||
|
|
'num_heads': [4, 8, 16, 32],
|
||
|
|
'patch_size': (2, 4, 4),
|
||
|
|
'window_size': (16, 7, 7),
|
||
|
|
'drop_path_rate': 0.4,
|
||
|
|
'patch_norm': True},
|
||
|
|
}
|
||
|
|
return args[model_name]
|
||
|
|
|