From 877dbb338c1c2031123c7c0992c75e1ce364e4c1 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 10 Feb 2023 18:19:32 +0800 Subject: [PATCH] Update device Signed-off-by: Jael Gu --- nn_fingerprint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nn_fingerprint.py b/nn_fingerprint.py index 46e6c7d..e7f8bb7 100644 --- 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)