diff --git a/IGEV-Stereo/evaluate_stereo.py b/IGEV-Stereo/evaluate_stereo.py index 73d35dd..e7ff54a 100644 --- a/IGEV-Stereo/evaluate_stereo.py +++ b/IGEV-Stereo/evaluate_stereo.py @@ -169,7 +169,7 @@ def validate_sceneflow(model, iters=32, mixed_prec=False): @torch.no_grad() -def validate_middlebury(model, iters=32, split='F', mixed_prec=False): +def validate_middlebury(model, iters=32, split='H', mixed_prec=False): """ Peform validation using the Middlebury-V3 dataset """ model.eval() aug_params = {}