From 4aa5e0d91edf92d7c018ba1f6041f6964b184575 Mon Sep 17 00:00:00 2001
From: HTensor <27774381+HTensor@users.noreply.github.com>
Date: Sat, 29 Apr 2023 23:24:10 +0800
Subject: [PATCH] compatibility changes

---
 IGEV-Stereo/demo_imgs.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/IGEV-Stereo/demo_imgs.py b/IGEV-Stereo/demo_imgs.py
index e7f8a10..094b5af 100644
--- a/IGEV-Stereo/demo_imgs.py
+++ b/IGEV-Stereo/demo_imgs.py
@@ -88,6 +88,7 @@ if __name__ == '__main__':
     parser.add_argument('--slow_fast_gru', action='store_true', help="iterate the low-res GRUs more frequently")
     parser.add_argument('--n_gru_layers', type=int, default=3, help="number of hidden GRU levels")
     parser.add_argument('--max_disp', type=int, default=192, help="max disp of geometry encoding volume")
+    parser.add_argument('--freeze_backbone_params', action="store_true", help="freeze backbone parameters")
 
     args = parser.parse_args()