Update device

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