Update kernel_utils.py

kernel_utils.py

@@ -325,7 +325,7 @@ def predict_on_video(face_extractor, video_path, batch_size, input_size, models,
                 x = torch.tensor(x, device="cpu").float()
                 # Preprocess the images.
                 x = x.permute((0, 3, 1, 2))
-                x = x.to('cpu')
+                # x = x.to('cpu')
                 for i in range(len(x)):
                     x[i] = normalize_transform(x[i] / 255.)
                 # Make a prediction, then take the average.
@@ -333,6 +333,7 @@ def predict_on_video(face_extractor, video_path, batch_size, input_size, models,
                     preds = []
                     for model in models:
                         # with torch.cuda.amp.autocast():
+                        model = model.float()
                         y_pred = model(x[:n].float())
                         y_pred = torch.sigmoid(y_pred.squeeze())
                         bpred = y_pred[:n].cpu().numpy()