diff --git a/inference_video.py b/inference_video.py index 8f1d095..355d9f2 100644 --- a/inference_video.py +++ b/inference_video.py @@ -13,7 +13,6 @@ if torch.cuda.is_available(): torch.set_grad_enabled(False) torch.backends.cudnn.enabled = True torch.backends.cudnn.benchmark = True - torch.set_num_threads(4) parser = argparse.ArgumentParser(description='Interpolation for a pair of images') parser.add_argument('--video', dest='video', required=True) diff --git a/inference_video_parallel.py b/inference_video_parallel.py index eaf18a1..becdfd5 100644 --- a/inference_video_parallel.py +++ b/inference_video_parallel.py @@ -13,7 +13,6 @@ if torch.cuda.is_available(): torch.set_grad_enabled(False) torch.backends.cudnn.enabled = True torch.backends.cudnn.benchmark = True - torch.set_num_threads(4) parser = argparse.ArgumentParser(description='Interpolation for a pair of images') parser.add_argument('--video', dest='video', required=True)