Discriminator Class Reference

#include <network.hpp>

Inheritance diagram for Discriminator:

Public Member Functions
	Discriminator (int nc_=3, int ngf_=64, int ndf_=64)
torch::nn::Sequential	get_module ()

Detailed Description

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.

Constructor & Destructor Documentation

Discriminator()

Discriminator::Discriminator ( int  nc_ = 3, int  ngf_ = 64, int  ndf_ = 64  )

inline

Member Function Documentation

get_module()

torch::nn::Sequential Discriminator::get_module ( )

inline

---

The documentation for this class was generated from the following file:

• include/network.hpp