fix: tensor operation bugfix

lbhd/infer.py

@@ -65,12 +65,12 @@ def videofile_to_scores(videofile, model):
     frames = videofile_to_frames(videofile)
     x = frames_to_vectors(frames, model.frames_per_clip)
     with torch.no_grad():
-        return torch.sigmoid(model(x)).squeeze().numpy()
+        return torch.sigmoid(model(x)).squeeze().cpu().numpy()
 
 def frames_to_scores(frames, model, use_frame_diff):
     x = frames_to_vectors(frames, model.frames_per_clip, use_frame_diff)
     with torch.no_grad():
-        return torch.sigmoid(model(x)).squeeze().numpy()
+        return torch.sigmoid(model(x)).squeeze().cpu().numpy()
 
 def prepare_output(scores):
     output = [[clip_id, clip_id+2, s] for (clip_id, s) in enumerate(scores)]