#include <network.hpp>
A quick note on Discriminator in DCGAN:
- Spatial Pooling Layers such as MaxPool layers were replaced with Strided Convolutions.
- Use BatchNorm in the Discriminator.
- Remove FC layers for deeper architectures.
- Use LeakyReLU activation function for all the layers in the Discriminator.
◆ Discriminator()
Discriminator::Discriminator |
( |
int |
nc_ = 3 , |
|
|
int |
ngf_ = 64 , |
|
|
int |
ndf_ = 64 |
|
) |
| |
|
inline |
◆ get_module()
torch::nn::Sequential Discriminator::get_module |
( |
| ) |
|
|
inline |
The documentation for this class was generated from the following file: