Function load_checkpoin

VAE_AutoEncoder/utils.py:35–42

(checkpoint_file, model, optimizer, lr)

def load_checkpoin(checkpoint_file, model, optimizer, lr):
    print("=> Loading checkpoint")
    checkpoint = torch.load(checkpoint_file, map_location=config.DEVICE)
    model.load_state_dict(checkpoint["state_dict"])
    optimizer.load_state_dict(checkpoint["optimizer"])

    for param_group in optimizer.param_group:
        param_group["lr"] = lr

def generate_and_save_images(save_dir,features,gen,epoch):
    """