From 93371a3c4d76dc43eaeb0ad278eb42e69f3ad880 Mon Sep 17 00:00:00 2001
From: HTensor <27774381+HTensor@users.noreply.github.com>
Date: Wed, 19 Apr 2023 00:58:31 +0800
Subject: [PATCH] todo

---
 IGEV-Stereo/train_stereo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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))