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

Public Member Functions

 Arguments (std::string data_root, int num_workers, int bs, int img_size, int num_channels, int length_latent_vector, int depth_feature_maps_g, int depth_feature_maps_d, int number_epochs, float learning_rate, float beta_1, int num_gpu)
 

Public Attributes

std::string dataroot
 
int workers
 
int batch_size
 
int image_size
 
int nc
 
int nz
 
int ngf
 
int ndf
 
int num_epochs
 
float lr
 
float beta1
 
int ngpu
 

Detailed Description

Arguments Class: Makes loading arguments for training easy!

Constructor & Destructor Documentation

◆ Arguments()

Arguments::Arguments ( std::string  data_root,
int  num_workers,
int  bs,
int  img_size,
int  num_channels,
int  length_latent_vector,
int  depth_feature_maps_g,
int  depth_feature_maps_d,
int  number_epochs,
float  learning_rate,
float  beta_1,
int  num_gpu 
)
inline

Member Data Documentation

◆ batch_size

int Arguments::batch_size

◆ beta1

float Arguments::beta1

◆ dataroot

std::string Arguments::dataroot

◆ image_size

int Arguments::image_size

◆ lr

float Arguments::lr

◆ nc

int Arguments::nc

◆ ndf

int Arguments::ndf

◆ ngf

int Arguments::ngf

◆ ngpu

int Arguments::ngpu

◆ num_epochs

int Arguments::num_epochs

◆ nz

int Arguments::nz

◆ workers

int Arguments::workers

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