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`):