##checkpoints are loaded successfully and all keys match successfully
print("=> Loading checkpoint")
checkpoint = torch.load(’/content/High/generator_C.pth’, map_location=“cuda”)
gen.load_state_dict(checkpoint[‘state_dict’])

now feeding noise to the generator t get the desired image
alpha = 1.0
for i in range(10):
noise =torch.randn((1, Z_DIM, 1, 1))
img_size=128
num_steps = int(log2(img_size / 4))
img = gen(noise, alpha, steps=num_steps)
save_image(img*0.5+0.5, f"/content/img_{i}.png")
###but the output image is’