From 9c74ae6175d43e5b93c72c63efeff6df50d0c263 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 17 Jun 2022 10:56:34 +0800 Subject: [PATCH] Remove cuda name typo Signed-off-by: Jael Gu --- action_clip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/action_clip.py b/action_clip.py index 409cd27..4660583 100644 --- 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