[Image] 이미지 데이터 확인할 때 개수가 달라지는 이유

6 x 6 36개로 출력됐는데 데이터 셋을 변경하니 13개로 개수가 줄었음

 

데이터셋 개수와 batch size와 관련이 있음

 

데이터 셋이 100개 batch size가 33 이라고 한다면

33, 33, 33, 1 이렇게 갈라져서 마지막 output이 1이기 때문에 이미지가 1개만 나옴

batch size를 알맞게 변경해주면 됨

 

 

    if (epoch+1) % config.save_interval == 0:
        save_path = os.path.join(config.save_path, config.dataset, 'epoch_[{}].png'.format(epoch+1))
        gen_img = config.denormalize(gen_img)
        torchvision.utils.save_image(gen_img.data[:16], save_path, nrow=4, normalize=True)

---

torch.Size를 써서 해결하는 방법도 있음