How do I show a picture like you that is all in one category？

[LittleStars666]

The pictures I trained showed ten different categories of a picture，but I want to save them one by one. Is there any way ？

[Hydrino]

labels = 4*torch.ones((10,),dtype = torch.long, device = device,) fakes = gen(noise,labels).detach() showImage(make_grid(fakes))

replace the '4' with the category you want to show pictures of.

[LittleStars666]

I tried, and the result was the same