Function save_checkpoint

Code/utils.py:16–22

(model,optimizer,filename = "my_checkpoint.pth.tar",epochs = 0)

import numpy as np

def save_checkpoint(model,optimizer,filename = "my_checkpoint.pth.tar",epochs = 0):
    print("=> Saving checkpoint")
    checkpoint = {
        "state_dict":model.state_dict(),
        "optimizer":optimizer.state_dict(),
    }
    torch.save(checkpoint,filename + str(epochs))
def load_checkpoin(checkpoint_file,model,optimizer,lr):
    print("=> Loading checkpoint")
    checkpoint = torch.load(checkpoint_file,map_location=config.DEVICE)