Update pre.py

ZhuLvs · Nov 27, 2024 · b8d0ee1 · b8d0ee1
1 parent c556b0d
commit b8d0ee1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/pre.py b/model/pre.py
@@ -38,7 +38,7 @@ def remove_module_prefix(state_dict):
     model = CustomDeepLabV3().to(device)
 
     # Load the state dict with 'module.' prefix removed
-    state_dict = torch.load("../result/best_model.pth", map_location=device)
+    state_dict = torch.load("./result/best_model.pth", map_location=device)
     new_state_dict = remove_module_prefix(state_dict)
     model.load_state_dict(new_state_dict)