wgan gp

WGAN-GP

Uma implementação de Pytorch do artigo "Treinamento aprimorado de Wasserstein Gans".

Pré -requisitos

Python, Numpy, Scipy, Matplotlib Uma GPU NVIDIA recente

Uma versão mestre mais recente do pytorch

Progresso

• gan_toy.py: conjuntos de dados de brinquedos (8 gaussianos, 25 gaussianos, rolo suíço). ( Concluído em 2017.5.8)

• gan_language.py: Modelo de linguagem no nível do caractere (o discriminador está usando o gerador NN.Conv1d . Está usando o nn.conv1d .

• gan_mnist.py: MNIST ( os resultados em execução enquanto terminaram em 2017.6.26. O discriminador está usando o NN.Conv1d . O gerador está usando o nn.conv1d .)

• gan_64x64.py: 64x64 Arquiteturas ( ansiosas pela sua solicitação de tração )

• gan_cifar.py: cifar-10 ( ótimo obrigado ao robotcator )

Resultados

• Conjunto de dados de brinquedos

  Algum resultado de amostra, você pode consultar os resultados/ brinquedo/ pasta para obter detalhes .

  • 8Gaussians 154500 iteração

  

  • 25Gaussians 48500 iteração

    

  • Swissroll 69400 iteração

  

• Conjunto de dados mnist

  Algum resultado de amostra, você pode consultar os resultados/ mnist/ pasta para obter detalhes .

  

  

  

  

• Bilhão de geração de linguagem de palavras (usando CNN, nível de caractere)

  Algum resultado de amostra após 8699 épocas, que são detalhadas na amostra

  Eu não corro épocas suficientes, pois isso é muito difícil.

  Ele moveu o tapete, clice T

  Um pouco de marchas forreuid ele

  Com o deputado esfarrapado

  PAMREUTOL O RENCED EM CARRO INSON

  Nem op aos ficomens lecs o

  Em é um "norleado pelo OT pode

  O Onteon eu preto este pirder,

  É a aoracia de "Medurn

  Rame ele reapareceu para thim atreast

  O stinl que herth of the não t

  O idiota é o que você está em nalência

  É um the tose sho leloch cumm

• Conjunto de dados CIFAR10

  Algum resultado da amostra, você pode consultar os resultados/ CIFAR10/ PASTER para obter detalhes .

  

Reconhecer

Baseado na implementação IGUL222/aprimored_wgan_training e Martinarjovsky/wassersteingan