From 90b3497440f0f21b037eef00ed7c78415e3417b1 Mon Sep 17 00:00:00 2001 From: xujinling Date: Mon, 13 Jun 2022 14:21:20 +0800 Subject: [PATCH] add2 Signed-off-by: xujinling --- __init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/__init__.py b/__init__.py index 4191b3c..d59966f 100644 --- a/__init__.py +++ b/__init__.py @@ -15,6 +15,6 @@ from .video_swin_transformer import VideoSwinTransformer -def video_swin_transformer(model_name: str, modality: str, **kwargs): - return VideoSwinTransformer(model_name, modality, **kwargs) +def video_swin_transformer(**kwargs): + return VideoSwinTransformer(**kwargs)