Skip to content
Snippets Groups Projects
Commit b24dd302 authored by hsfzxjy's avatar hsfzxjy
Browse files

Fix bug: remove unecessary assertion

parent 38e792c2
No related branches found
No related tags found
No related merge requests found
......@@ -82,7 +82,6 @@ def main():
pretrained_dict = torch.load(model_state_file)
model_dict = model.state_dict()
assert set(k[6:] for k in pretrained_dict) == set(model_dict)
pretrained_dict = {k[6:]: v for k, v in pretrained_dict.items()
if k[6:] in model_dict.keys()}
for k, _ in pretrained_dict.items():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment