torch-GAN/example.py at main · SashaMatsun/torch-GAN · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from torch_gan import Model
from torch_gan.generators import DCGANGenerator
from torch_gan.discriminators import DCGANDiscriminator
from torch_gan.generators import CGANGenerator
from torch_gan.discriminators import CGANDiscriminator
from torch_gan.generators import P2PGenerator
from torch_gan.discriminators import P2PDiscriminator
from torch_gan.discriminators import InfoGANDiscriminator
from torch_gan.discriminators import BEGANDiscriminator
from torch_gan.generators import SAGANGenerator
from torch_gan.discriminators import SAGANDiscriminator
from torch_gan.generators import ProGANGenerator
from torch_gan.discriminators import ProGANDiscriminator


g = SAGANGenerator(64, 128, 3)
d = SAGANDiscriminator(64, 3)

model = Model(g, d)