Update demo_dense_visualize.py

demo_dense_visualize.py

@@ -107,7 +107,7 @@ class Tracker:
             inference_iters: Number of inference iterations.
             device: Torch device, defaults to 'cuda'.
         """
-        self.model = model
+        self.model = model.cuda()
         self.S = S
         self.stride = stride
         self.inference_iters = inference_iters