Remove cuda name typo

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · 9c74ae6175
1 changed files with 1 additions and 1 deletions
--- a/action_clip.py
+++ b/action_clip.py
@ -42,7 +42,7 @@ class ActionClip(NNOperator):
                 topk: int = 5
                 ):
        super().__init__(framework='pytorch')
-        self.device = 'cuda:2' if torch.cuda.is_available() else 'cpu'
+        self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
        self.model_name = model_name
        self.skip_preprocess = skip_preprocess
        self.topk = topk