fix device setting

models/hyperstyle/hypernetworks/hypernetwork.py

@@ -91,7 +91,7 @@ class SharedWeightsHyperNetResNetSeparable(Module):
             self.layers_to_tune = [int(l) for l in opts.layers_to_tune.split(',')]
 
         self.shared_layers = [0, 2, 3, 5, 6, 8, 9, 11, 12]
-        self.shared_weight_hypernet = SharedWeightsHypernet(in_size=512, out_size=512, mode=None)
+        self.shared_weight_hypernet = SharedWeightsHypernet(in_size=512, out_size=512, mode=None, device=opts.device)
 
         self.refinement_blocks = nn.ModuleList()
         self.n_outputs = opts.n_hypernet_outputs

models/hyperstyle/utils/model_utils.py

@@ -32,7 +32,7 @@ def load_model(checkpoint_path, device='cuda', update_opts=None, is_restyle_enco
         net = HyperStyle(opts)
 
     net.eval()
-    net.to(device)
+    net.to(opts.device)
     return net, opts