diff --git a/IGEV-Stereo/train_stereo.py b/IGEV-Stereo/train_stereo.py
index 4b8b115..92c9981 100644
--- a/IGEV-Stereo/train_stereo.py
+++ b/IGEV-Stereo/train_stereo.py
@@ -132,7 +132,7 @@ class Logger:
 
 
 def train(args):
-
+    # todo: compile the model to speed up at pytorch 2.0.
     model = nn.DataParallel(IGEVStereo(args))
     print("Parameter Count: %d" % count_parameters(model))