fix #835: some broken imports

This commit is contained in:
源文雨
2023-07-23 14:32:53 +08:00
parent a002f817df
commit 18067aa85d
5 changed files with 6 additions and 6 deletions

View File

@@ -44,9 +44,10 @@ def load_checkpoint_d(checkpoint_path, combd, sbd, optimizer=None, load_opt=1):
model.module.load_state_dict(new_state_dict, strict=False)
else:
model.load_state_dict(new_state_dict, strict=False)
return model
go(combd, "combd")
go(sbd, "sbd")
model = go(sbd, "sbd")
#############
logger.info("Loaded model weights")