l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
2 years ago · 5f48516dc1
1 changed files with 1 additions and 1 deletions
--- 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,