Browse Source
Remove cuda name typo
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
action_clip.py
|
@ -42,7 +42,7 @@ class ActionClip(NNOperator): |
|
|
topk: int = 5 |
|
|
topk: int = 5 |
|
|
): |
|
|
): |
|
|
super().__init__(framework='pytorch') |
|
|
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.model_name = model_name |
|
|
self.skip_preprocess = skip_preprocess |
|
|
self.skip_preprocess = skip_preprocess |
|
|
self.topk = topk |
|
|
self.topk = topk |
|
|