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'