logo
Browse Source

modify weights name

Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling 2 years ago
parent
commit
19bac24ffe
  1. 2
      video_swin_transformer.py

2
video_swin_transformer.py

@ -23,7 +23,7 @@ class VideoSwinTransformer(NNOperator):
Args:
model_name (`str`):
Supported model names:
- swin_t_w877_k400_1k
- swin_t_k400_1k
skip_preprocess (`str`):
Flag to skip video transforms.
classmap (`str=None`):

Loading…
Cancel
Save