Skip to content

Commit f6513da

Browse files
committed
changed strict loading to false
1 parent 48bb5e1 commit f6513da

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bossnet/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def _load_model(self):
312312
"""
313313
model_path = os.path.join(os.path.split(os.path.abspath(__file__))[0], "deconstructed_model")
314314
state_dict = franken_load(model_path, 10)
315-
self.model.load_state_dict(state_dict, strict=True)
315+
self.model.load_state_dict(state_dict, strict=False)
316316
self.model.to(self.device)
317317
self.model.eval()
318318

0 commit comments

Comments
 (0)