Mirrors/the-algorithm-ml
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm-ml.git synced 2025-01-09 22:39:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

use else blocks

Browse Source
This commit is contained in:
rickdeu 2023-04-02 21:24:55 +01:00
parent 78c3235eee
commit d02aa2ad8f
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) snapshot.restore(self.state)
# we still need to ensure that extra_state has walltime in it # we still need to ensure that extra_state has walltime in it
self.state["extra_state"] = torchsnapshot.StateDict(step=self.step, walltime=0.0) self.state["extra_state"] = torchsnapshot.StateDict(step=self.step, walltime=0.0)
else:
logging.info(f"Restored snapshot from {snapshot.path}. ({time.time() - start_time:.05}s") logging.info(f"Restored snapshot from {snapshot.path}. ({time.time() - start_time:.05}s")
@classmethod @classmethod
def get_torch_snapshot( def get_torch_snapshot(