from local

This commit is contained in:
dohe0342 2022-12-30 11:17:13 +09:00
parent 6f1a16af61
commit e29b45acb1
2 changed files with 1 additions and 0 deletions

View File

@ -122,6 +122,7 @@ def load_checkpoint(
"""
TODO: document it
"""
print(model)
logging.info(f"Loading checkpoint from {filename}")
checkpoint = torch.load(filename, map_location="cpu")