From fd0f334da777e6ece690d1015b0c09939a7bf4bb Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Fri, 20 Nov 2020 15:08:52 +0800
Subject: [PATCH] Remove thread setting

---
 inference_video.py          | 1 -
 inference_video_parallel.py | 1 -
 2 files changed, 2 deletions(-)

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)