hanbit:

Is there a faster or more ‘data-parallel’ way to save i


I am using matplot pyplot, I am not sure what I am doing wrong. it always print the first image. Any help?
for t in range(0,batch_size):
    a = image[t].cpu().numpy()
    plt.savefig("./test.png")
    plt.imshow(a)