todo

2023-04-19 00:58:31 +08:00 · 2023-04-19 00:58:31 +08:00 · 93371a3c4d
commit 93371a3c4d
parent 54a2eabbff
1 changed files with 1 additions and 1 deletions
--- 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))