From ba4c7e339bae1b2fedd1f1be89e1ce7afb54e566 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 a8aa3c8..ad0b8b5 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'