Mirrors/the-algorithm-ml
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm-ml.git synced 2024-06-04 16:58:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge d02aa2ad8f into b85210863f

Browse Source
This commit is contained in:
André Hangalo 2023-04-06 10:57:54 +08:00 committed by GitHub
commit d674e724d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
common/checkpointing/snapshot.py
View File

@ -73,8 +73,8 @@ class Snapshot:
snapshot.restore(self.state)
# we still need to ensure that extra_state has walltime in it
self.state["extra_state"] = torchsnapshot.StateDict(step=self.step, walltime=0.0)
logging.info(f"Restored snapshot from {snapshot.path}. ({time.time() - start_time:.05}s")
else:
logging.info(f"Restored snapshot from {snapshot.path}. ({time.time() - start_time:.05}s")
@classmethod
def get_torch_snapshot(