From 19bac24ffeed380e0691bbafdce51a949d84e14d Mon Sep 17 00:00:00 2001 From: xujinling Date: Wed, 15 Jun 2022 10:54:17 +0800 Subject: [PATCH] modify weights name Signed-off-by: xujinling --- video_swin_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/video_swin_transformer.py b/video_swin_transformer.py index 2f65cf5..250e475 100644 --- a/video_swin_transformer.py +++ b/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`):