From 5f48516dc1fdc69992cb4a51ff45b9f5ceab53db Mon Sep 17 00:00:00 2001
From: xujinling <jinling.xu@zilliz.com>
Date: Wed, 15 Jun 2022 19:14:48 +0800
Subject: [PATCH] l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
---
 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 db5d0f5..9a1c67f 100644
--- a/video_swin_transformer.py
+++ b/video_swin_transformer.py
@@ -57,7 +57,7 @@ class VideoSwinTransformer(NNOperator):
         self.model = video_swin_transformer.create_model(model_name=self.model_name,
                                                          pretrained=True,
                                                          device=self.device)
-
+        self.model.to(self.device)
         self.transform_cfgs = get_configs(
             side_size=224,
             crop_size=224,