Browse Source
modify weights name
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
2 years ago
1 changed files with
1 additions and
1 deletions
-
video_swin_transformer.py
|
|
@ -23,7 +23,7 @@ class VideoSwinTransformer(NNOperator): |
|
|
Args: |
|
|
Args: |
|
|
model_name (`str`): |
|
|
model_name (`str`): |
|
|
Supported model names: |
|
|
Supported model names: |
|
|
- swin_t_w877_k400_1k |
|
|
|
|
|
|
|
|
- swin_t_k400_1k |
|
|
skip_preprocess (`str`): |
|
|
skip_preprocess (`str`): |
|
|
Flag to skip video transforms. |
|
|
Flag to skip video transforms. |
|
|
classmap (`str=None`): |
|
|
classmap (`str=None`): |
|
|
|
