DCGAN on CelebA Dataset using Libtorch (PyTorch C++ Frontend API)
Public Member Functions | List of all members
Generator Class Reference

#include <network.hpp>

Inheritance diagram for Generator:

Public Member Functions

 Generator (int nc_=3, int nz_=100, int ngf_=64)
 
torch::nn::Sequential get_module ()
 

Detailed Description

Generator consists: ConvTranspose2d layers, BatchNorm layers, ReLU Activations Input to Generator: latent vector, z, drawn from standard normal distribution Output is 3x64x64 RGB Image

Constructor & Destructor Documentation

◆ Generator()

Generator::Generator ( int  nc_ = 3,
int  nz_ = 100,
int  ngf_ = 64 
)
inline

Member Function Documentation

◆ get_module()

torch::nn::Sequential Generator::get_module ( )
inline

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