From 54442acefe8e05b73be5dd6208dd6118a1064f6a Mon Sep 17 00:00:00 2001 From: Sujeevan Rasaratnam Date: Fri, 24 May 2024 19:45:14 +0000 Subject: [PATCH] don't use rocm variant of torchvision --- data/video_to_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/video_to_image.py b/data/video_to_image.py index 2fdf2f8c..40297f19 100644 --- a/data/video_to_image.py +++ b/data/video_to_image.py @@ -35,7 +35,7 @@ class VideoToFrames: Additionally detects all faces for each frame. """ - _modules_needed_by_name = ["torchvision==0.18.0+rocm6.0", "facenet-pytorch==2.5.3"] + _modules_needed_by_name = ["torchvision==0.18.0", "facenet-pytorch==2.5.3"] def __init__(self, num_frames_per_video=3, face_additional_area=0.5): self.num_frames_per_video = num_frames_per_video