Browse Source
l
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
2 years ago
1 changed files with
1 additions and
1 deletions
-
video_swin_transformer.py
|
|
@ -57,7 +57,7 @@ class VideoSwinTransformer(NNOperator): |
|
|
self.model = video_swin_transformer.create_model(model_name=self.model_name, |
|
|
self.model = video_swin_transformer.create_model(model_name=self.model_name, |
|
|
pretrained=True, |
|
|
pretrained=True, |
|
|
device=self.device) |
|
|
device=self.device) |
|
|
|
|
|
|
|
|
|
|
|
self.model.to(self.device) |
|
|
self.transform_cfgs = get_configs( |
|
|
self.transform_cfgs = get_configs( |
|
|
side_size=224, |
|
|
side_size=224, |
|
|
crop_size=224, |
|
|
crop_size=224, |
|
|
|
