Update device

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · cb8fb811d7
1 changed files with 1 additions and 1 deletions
--- a/nn_fingerprint.py
+++ b/nn_fingerprint.py
@ -69,7 +69,7 @@ class Model:
    def __call__(self, *args, **kwargs):
        new_args = []
        new_kwargs = {}
-        for x in new_args:
+        for x in args:
            new_args.append(x.to(self.device))
        for k, v in kwargs.items():
            new_kwargs[k] = v.to(self.device)