Update common.py

XiandaGuo · Apr 3, 2024 · b9d1f45 · b9d1f45
1 parent e3dbc69
commit b9d1f45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openstereo/utils/common.py b/openstereo/utils/common.py
@@ -234,7 +234,7 @@ def convert_state_dict(ori_state_dict, is_dist=True):
             new_state_dict = ori_state_dict
         else:
             for k, v in ori_state_dict.items():
-                k = k.replace('module.', '')
+                k = k[7:]
                 new_state_dict[k] = v
 
     return new_state_dict