Insufficient video memory

[userLx888]

Training structure_generator.In order to solve the problem of insufficient video memory during generator, reducing batch_size leads to an error when saving the training results : Index 1 is out of bounds for axis 0 with size 1.I hope I can get your help. Thank you!

[USTC-JialunPeng]

What is your batch size? And please attach your error log.

[userLx888]

my batch size is 2,

[USTC-JialunPeng]

I found the problem. The visualization of structure feature maps shows 4 images. You can replace line162 with height = size and line167 with for i in range(1): to show only 1 image. https://github.com/USTC-JialunPeng/Diverse-Structure-Inpainting/blob/0def26df30f291cdfd68c248228366f9d0a8f7e5/net/nn.py#L155-L179

[userLx888]

OK OK,I get it ,thank you very much!

[userLx888]

OK OK,I get it ,thank you very much!