disable cudnn.benchmark

2024-04-23 11:51:21 +03:00
parent af5378d952
commit c6c4135e1b
1 changed files with 1 additions and 1 deletions
@@ -25,7 +25,7 @@ from .models.FLAME import FLAME, FLAMETex
 from .utils import util
 from .utils.tensor_cropper import transform_points
 from skimage.io import imread
-torch.backends.cudnn.benchmark = True
+#torch.backends.cudnn.benchmark = True
 import numpy as np

 def recursive_to(x, target):