Try to improve speed?

test.py

@@ -5,6 +5,11 @@ import cv2
 import numpy as np
 import torch
 import architecture as arch
+import multiprocessing
+
+cpu_count = multiprocessing.cpu_count()
+
+torch.set_num_threads(cpu_count)
 
 model_path = '4x_eula_digimanga_bw_v2_nc1_307k.pth'
 img_path = sys.argv[1]