|
| 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) |
|
Arguments Class: Makes loading arguments for training easy!
◆ 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 |
◆ batch_size
int Arguments::batch_size |
◆ beta1
◆ dataroot
std::string Arguments::dataroot |
◆ image_size
int Arguments::image_size |
◆ lr
◆ nc
◆ ndf
◆ ngf
◆ ngpu
◆ num_epochs
int Arguments::num_epochs |
◆ nz
◆ workers
The documentation for this class was generated from the following file: