from local
This commit is contained in:
parent
6f1a16af61
commit
e29b45acb1
Binary file not shown.
@ -122,6 +122,7 @@ def load_checkpoint(
|
|||||||
"""
|
"""
|
||||||
TODO: document it
|
TODO: document it
|
||||||
"""
|
"""
|
||||||
|
print(model)
|
||||||
logging.info(f"Loading checkpoint from {filename}")
|
logging.info(f"Loading checkpoint from {filename}")
|
||||||
checkpoint = torch.load(filename, map_location="cpu")
|
checkpoint = torch.load(filename, map_location="cpu")
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user