self

Signed-off-by: xujinling <jinling.xu@zilliz.com>
3 years ago · 8d3acfef12
1 changed files with 1 additions and 1 deletions
--- a/frozen_in_time.py
+++ b/frozen_in_time.py
@ -67,7 +67,7 @@ class FrozenInTime(NNOperator):
                                                 video_is_load_pretrained=False,
                                                 video_model_type='SpaceTimeTransformer',
                                                 text_is_load_pretrained=False,
-                                                 device=device)
+                                                 device=self.device)

        self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False)
        self.transform_cfgs = get_configs(