msurtsukov / generative-models-collection Goto Github PK

View Code? Open in Web Editor NEW

35.0 3.0 7.0 154.81 MB

PyTorch implemented generative models for CelebA dataset: DCGAN, LSGAN, WGAN, WGANGP, InfoGAN, BEGAN, VAE, VAEGAN

License: MIT License

Python 100.00%

python pytorch gan dcgan vae vaegan wgan wgangp began infogan

generative-models-collection's Issues

cannot find list_attr_celeba.csv in dataset

Thanks for you code, i cannot find list_attr_celeba.csv in dataset. Where it form?

What is Gamma Refer to?

Hi,

Please see https://github.com/msurtsukov/generative-models-collection#vaegan.

expanded_attrs = attrs.view(attrs.shape, 1, 1).expand(attrs.shape, w, h) seems to be a problem

    if self.n_attrs:
        batch_size, c, w, h = x.shape
        expanded_attrs = attrs.view(
            *attrs.shape, 1, 1).expand(*attrs.shape, w, h)
        x = torch.cat((x, expanded_attrs), dim=1)

x = torch.cat((x, expanded_attrs), dim=1)
RuntimeError: Tensors must have same number of dimensions: got 4 and 6

Hello, when I run this program （VAEGAN）, these sentences of code will have problems. The original data is 4-dimensional, and the torch cannot be executed after dimension upgrading or expansion torch.cat operation

Recommend Projects