From 3b68318ed59ad315d140ab8ec5ec9e70bdea1436 Mon Sep 17 00:00:00 2001 From: HTensor <27774381+HTensor@users.noreply.github.com> Date: Wed, 3 May 2023 04:13:28 +0800 Subject: [PATCH] disabled cudnn benchmark --- IGEV-Stereo/train_stereo.py | 1 - 1 file changed, 1 deletion(-) diff --git a/IGEV-Stereo/train_stereo.py b/IGEV-Stereo/train_stereo.py index feaf353..98a0f62 100644 --- a/IGEV-Stereo/train_stereo.py +++ b/IGEV-Stereo/train_stereo.py @@ -22,7 +22,6 @@ from evaluate_stereo import * import core.stereo_datasets as datasets import torch.nn.functional as F -torch.backends.cudnn.benchmark = True ckpt_path = './checkpoints/igev_stereo' log_path = './checkpoints/igev_stereo'