diff --git a/video_swin_transformer.py b/video_swin_transformer.py index a831908..75ab8c6 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_tiny_patch244_window877_kinetics400_1k + - swin_t_w877_k400_1k skip_preprocess (`str`): Flag to skip video transforms. classmap (`str=None`): @@ -32,7 +32,7 @@ class VideoSwinTransformer(NNOperator): The number of classification labels to be returned (ordered by possibility from high to low). """ def __init__(self, - model_name: str = 'swin_tiny_patch244_window877_kinetics400_1k', + model_name: str = 'swin_t_w877_k400_1k', framework: str = 'pytorch', skip_preprocess: bool = False, classmap: str = None,