diff --git a/IGEV-Stereo/core/stereo_datasets.py b/IGEV-Stereo/core/stereo_datasets.py
index 21fc63f..20a4e83 100644
--- a/IGEV-Stereo/core/stereo_datasets.py
+++ b/IGEV-Stereo/core/stereo_datasets.py
@@ -73,7 +73,6 @@ class StereoDataset(data.Dataset):
         img2 = np.array(img2).astype(np.uint8)
 
         disp = np.array(disp).astype(np.float32)
-        assert not (True in np.isnan(disp))
 
         flow = np.stack([disp, np.zeros_like(disp)], axis=-1)