Skip to content

Instantly share code, notes, and snippets.

@FeepingCreature
Created July 9, 2018 17:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save FeepingCreature/319a9cc009705f111072b6a79340cb98 to your computer and use it in GitHub Desktop.
Save FeepingCreature/319a9cc009705f111072b6a79340cb98 to your computer and use it in GitHub Desktop.
DCGAN_G(
(main): Sequential(
(Start-ConvTranspose2d): ConvTranspose2d(128, 2048, kernel_size=(4, 4), stride=(1, 1), bias=False)
(Start-BatchNorm2d): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Start-ReLU): ReLU()
(Middle-UpSample [1]): Upsample(scale_factor=2, mode=nearest)
(Middle-Conv2d [1]): Conv2d(2048, 1024, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(Middle-BatchNorm2d [1]): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-ReLU [1]): ReLU()
(Middle-UpSample [2]): Upsample(scale_factor=2, mode=nearest)
(Middle-Conv2d [2]): Conv2d(1024, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(Middle-BatchNorm2d [2]): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-ReLU [2]): ReLU()
(Middle-UpSample [3]): Upsample(scale_factor=2, mode=nearest)
(Middle-Conv2d [3]): Conv2d(512, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(Middle-BatchNorm2d [3]): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-ReLU [3]): ReLU()
(Middle-UpSample [4]): Upsample(scale_factor=2, mode=nearest)
(Middle-Conv2d [4]): Conv2d(256, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(Middle-BatchNorm2d [4]): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-ReLU [4]): ReLU()
(End-UpSample): Upsample(scale_factor=2, mode=nearest)
(End-Conv2d): Conv2d(128, 3, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
(End-Tanh): Tanh()
)
)
DCGAN_D(
(main): Sequential(
(Start-Conv2d): Conv2d(3, 128, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False)
(Start-LeakyReLU): LeakyReLU(negative_slope=0.2, inplace)
(Middle-Conv2d [0]): Conv2d(128, 256, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False)
(Middle-BatchNorm2d [0]): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-LeakyReLU [0]): LeakyReLU(negative_slope=0.2, inplace)
(Middle-Conv2d [1]): Conv2d(256, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False)
(Middle-BatchNorm2d [1]): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-LeakyReLU [1]): LeakyReLU(negative_slope=0.2, inplace)
(Middle-Conv2d [2]): Conv2d(512, 1024, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False)
(Middle-BatchNorm2d [2]): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-LeakyReLU [2]): LeakyReLU(negative_slope=0.2, inplace)
(Middle-Conv2d [3]): Conv2d(1024, 2048, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1), bias=False)
(Middle-BatchNorm2d [3]): BatchNorm2d(2048, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(Middle-LeakyReLU [3]): LeakyReLU(negative_slope=0.2, inplace)
(End-Conv2d): Conv2d(2048, 1, kernel_size=(4, 4), stride=(1, 1), bias=False)
)
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment