"deploy/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "322efb3d719e7354097e8a1888696a823aca7fd3"
Unverified Commit 18617967 authored by Simon Lui's avatar Simon Lui Committed by GitHub
Browse files

Fix error message in model_patcher.py

Found while tinkering.
parent fe4c0740
...@@ -145,7 +145,7 @@ class ModelPatcher: ...@@ -145,7 +145,7 @@ class ModelPatcher:
model_sd = self.model_state_dict() model_sd = self.model_state_dict()
for key in self.patches: for key in self.patches:
if key not in model_sd: if key not in model_sd:
print("could not patch. key doesn't exist in model:", k) print("could not patch. key doesn't exist in model:", key)
continue continue
weight = model_sd[key] weight = model_sd[key]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment