diff --git a/IGEV-Stereo/train_stereo.py b/IGEV-Stereo/train_stereo.py
index 00e2deb..feaf353 100644
--- a/IGEV-Stereo/train_stereo.py
+++ b/IGEV-Stereo/train_stereo.py
@@ -68,9 +68,10 @@ def sequence_loss(disp_preds, disp_init_pred, disp_gt, valid, loss_gamma=0.9, ma
 
     metrics = {
         'epe': epe.mean().item(),
-        '1px': (epe < 1).float().mean().item(),
-        '3px': (epe < 3).float().mean().item(),
-        '5px': (epe < 5).float().mean().item(),
+        '0.5px': (epe < 0.5).float().mean().item(),
+        '1.0px': (epe < 1.0).float().mean().item(),
+        '2.0px': (epe < 2.0).float().mean().item(),
+        '4.0px': (epe < 4.0).float().mean().item(),
     }
 
     return disp_loss, metrics