From c6c4135e1bc37cab1af6495dc0156fff548b1217 Mon Sep 17 00:00:00 2001 From: Kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 23 Apr 2024 11:51:21 +0300 Subject: [PATCH] disable cudnn.benchmark --- motiondiff_modules/spectre/src/spectre.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/motiondiff_modules/spectre/src/spectre.py b/motiondiff_modules/spectre/src/spectre.py index 0a3043d..0242908 100644 --- a/motiondiff_modules/spectre/src/spectre.py +++ b/motiondiff_modules/spectre/src/spectre.py @@ -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):