Commit 8ab6cb09 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

added save_chekpoint function

parent 0a75ffb5
......@@ -239,8 +239,23 @@ class Solver():
def save_model(self):
pass
def save_checkpoint(self):
pass
def save_checkpoint(self, state, filename):
"""General Checkpoint Save
This function saves a general checkpoint for inference and/or resuming training
Args:
state (dict): Dictionary of all the relevant model components
Returns:
None
Raises:
None
"""
torch.save(state, filename)
def load_checkpoint(self):
pass
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment