Function save_checkpoint

StyleGAN/utils.py:58–64

(model, optimizer, filename="models/my_checkpoint.pth.tar")

#保存模型，通常的做法
def save_checkpoint(model, optimizer, filename="models/my_checkpoint.pth.tar"):
    print("=> Saving checkpoint")
    checkpoint = {
        "state_dict": model.state_dict(),
        "optimizer": optimizer.state_dict(),
    }
    torch.save(checkpoint, filename)


#加载模型