latest
Guides
Introduction
Installation
Quick Start
Visualizations
Tutorial
Introduction
Self-supervised GAN
Implementing Models
Discriminator
Generator
Training
Evaluation
Final Code
Conclusion
Baselines
API Reference
torch_mimicry.nets
torch_mimicry.nets.basemodel
torch_mimicry.nets.gan
Base Unconditional GAN
Base Conditional GAN
torch_mimicry.nets.dcgan
DCGAN CIFAR
DCGAN 32
DCGAN 48
DCGAN 64
DCGAN 128
DCGAN Base
torch_mimicry.nets.wgan_gp
WGAN-GP 32
WGAN-GP 48
WGAN-GP 64
WGAN-GP 128
WGAN-GP Base
torch_mimicry.nets.sngan
SNGAN 32
SNGAN 48
SNGAN 64
SNGAN 128
SNGAN Base
torch_mimicry.nets.cgan_pd
CGAN-PD 32
CGAN-PD 128
CGAN-PD Base
torch_mimicry.nets.ssgan
SSGAN 32
SSGAN 48
SSGAN 64
SSGAN 128
SSGAN Base
torch_mimicry.nets.infomax_gan
InfoMax-GAN 32
InfoMax-GAN 48
InfoMax-GAN 64
InfoMax-GAN 128
InfoMax-GAN Base
torch_mimicry.modules
Layers
Residual Blocks
Losses
torch_mimicry.training
Trainer
Logger
MetricLog
Scheduler
torch_mimicry.metrics
FID
KID
Inception Score
Metrics
torch_mimicry.datasets
Dataset Loaders
Image Loaders
torch_mimicry.utils
Common Utilities
torch_mimicry
Docs
»
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
A
add_metric() (MetricLog method)
B
BaseConditionalDiscriminator (class in torch_mimicry.nets.gan.cgan)
BaseConditionalGenerator (class in torch_mimicry.nets.gan.cgan)
BaseDiscriminator (class in torch_mimicry.nets.gan.gan)
(class in torch_mimicry.nets.infomax_gan.infomax_gan_base)
BaseGenerator (class in torch_mimicry.nets.gan.gan)
BaseModel (class in torch_mimicry.nets.basemodel.basemodel)
bottom_width (BaseConditionalGenerator attribute)
(BaseGenerator attribute)
(CGANPDBaseGenerator attribute)
(CGANPDGenerator128 attribute)
(CGANPDGenerator32 attribute)
(DCGANBaseGenerator attribute)
(DCGANGenerator128 attribute)
(DCGANGenerator32 attribute)
(DCGANGenerator48 attribute)
(DCGANGenerator64 attribute)
(DCGANGeneratorCIFAR attribute)
(InfoMaxGANBaseGenerator attribute)
(InfoMaxGANGenerator128 attribute)
(InfoMaxGANGenerator32 attribute)
(InfoMaxGANGenerator48 attribute)
(InfoMaxGANGenerator64 attribute)
(SNGANBaseGenerator attribute)
(SNGANGenerator128 attribute)
(SNGANGenerator32 attribute)
(SNGANGenerator48 attribute)
(SNGANGenerator64 attribute)
(SSGANBaseGenerator attribute)
(SSGANGenerator128 attribute)
(SSGANGenerator32 attribute)
(SSGANGenerator48 attribute)
(SSGANGenerator64 attribute)
(WGANGPBaseGenerator attribute)
(WGANGPGenerator128 attribute)
(WGANGPGenerator32 attribute)
(WGANGPGenerator48 attribute)
(WGANGPGenerator64 attribute)
C
CGANPDBaseDiscriminator (class in torch_mimicry.nets.cgan_pd.cgan_pd_base)
CGANPDBaseGenerator (class in torch_mimicry.nets.cgan_pd.cgan_pd_base)
CGANPDDiscriminator128 (class in torch_mimicry.nets.cgan_pd.cgan_pd_128)
CGANPDDiscriminator32 (class in torch_mimicry.nets.cgan_pd.cgan_pd_32)
CGANPDGenerator128 (class in torch_mimicry.nets.cgan_pd.cgan_pd_128)
CGANPDGenerator32 (class in torch_mimicry.nets.cgan_pd.cgan_pd_32)
close_writers() (Logger method)
compute_gan_loss() (BaseDiscriminator method)
(BaseGenerator method)
compute_gen_dist_feat() (in module torch_mimicry.metrics.compute_kid)
compute_gen_dist_stats() (in module torch_mimicry.metrics.compute_fid)
compute_gradient_penalty_loss() (WGANGPBaseDiscriminator method)
compute_infomax_loss() (BaseDiscriminator method)
compute_probs() (BaseDiscriminator method)
compute_real_dist_feat() (in module torch_mimicry.metrics.compute_kid)
compute_real_dist_stats() (in module torch_mimicry.metrics.compute_fid)
compute_ss_loss() (SSGANBaseDiscriminator method)
ConditionalBatchNorm2d (class in torch_mimicry.modules.layers)
count_params() (BaseModel method)
D
dataloader (Trainer attribute)
dataset_size (Logger attribute)
DBlock (class in torch_mimicry.modules.resblocks)
DBlockOptimized (class in torch_mimicry.modules.resblocks)
DCGANBaseDiscriminator (class in torch_mimicry.nets.dcgan.dcgan_base)
DCGANBaseGenerator (class in torch_mimicry.nets.dcgan.dcgan_base)
DCGANDiscriminator128 (class in torch_mimicry.nets.dcgan.dcgan_128)
DCGANDiscriminator32 (class in torch_mimicry.nets.dcgan.dcgan_32)
DCGANDiscriminator48 (class in torch_mimicry.nets.dcgan.dcgan_48)
DCGANDiscriminator64 (class in torch_mimicry.nets.dcgan.dcgan_64)
DCGANDiscriminatorCIFAR (class in torch_mimicry.nets.dcgan.dcgan_cifar)
DCGANGenerator128 (class in torch_mimicry.nets.dcgan.dcgan_128)
DCGANGenerator32 (class in torch_mimicry.nets.dcgan.dcgan_32)
DCGANGenerator48 (class in torch_mimicry.nets.dcgan.dcgan_48)
DCGANGenerator64 (class in torch_mimicry.nets.dcgan.dcgan_64)
DCGANGeneratorCIFAR (class in torch_mimicry.nets.dcgan.dcgan_cifar)
device (Logger attribute)
(Trainer attribute)
downsample (DBlock attribute)
E
evaluate() (in module torch_mimicry.metrics.compute_metrics)
F
fid_score() (in module torch_mimicry.metrics.compute_fid)
flush_secs (Logger attribute)
(Trainer attribute)
forward() (CGANPDDiscriminator128 method)
(CGANPDDiscriminator32 method)
(CGANPDGenerator128 method)
(CGANPDGenerator32 method)
(ConditionalBatchNorm2d method)
(DBlock method)
(DBlockOptimized method)
(DCGANDiscriminator128 method)
(DCGANDiscriminator32 method)
(DCGANDiscriminator48 method)
(DCGANDiscriminator64 method)
(DCGANDiscriminatorCIFAR method)
(DCGANGenerator128 method)
(DCGANGenerator32 method)
(DCGANGenerator48 method)
(DCGANGenerator64 method)
(DCGANGeneratorCIFAR method)
(GBlock method)
(InfoMaxGANDiscriminator128 method)
(InfoMaxGANDiscriminator32 method)
(InfoMaxGANDiscriminator48 method)
(InfoMaxGANDiscriminator64 method)
(InfoMaxGANGenerator128 method)
(InfoMaxGANGenerator32 method)
(InfoMaxGANGenerator48 method)
(InfoMaxGANGenerator64 method)
(SNGANDiscriminator128 method)
(SNGANDiscriminator32 method)
(SNGANDiscriminator48 method)
(SNGANDiscriminator64 method)
(SNGANGenerator128 method)
(SNGANGenerator32 method)
(SNGANGenerator48 method)
(SNGANGenerator64 method)
(SSGANDiscriminator128 method)
(SSGANDiscriminator32 method)
(SSGANDiscriminator48 method)
(SSGANDiscriminator64 method)
(SSGANGenerator128 method)
(SSGANGenerator32 method)
(SSGANGenerator48 method)
(SSGANGenerator64 method)
(SelfAttention method)
(WGANGPDiscriminator128 method)
(WGANGPDiscriminator32 method)
(WGANGPDiscriminator48 method)
(WGANGPDiscriminator64 method)
(WGANGPGenerator128 method)
(WGANGPGenerator32 method)
(WGANGPGenerator48 method)
(WGANGPGenerator64 method)
G
GBlock (class in torch_mimicry.modules.resblocks)
generate_images() (BaseConditionalGenerator method)
(BaseGenerator method)
generate_images_with_labels() (BaseConditionalGenerator method)
get_celeba_images() (in module torch_mimicry.datasets.image_loader)
get_cifar100_images() (in module torch_mimicry.datasets.image_loader)
get_cifar10_images() (in module torch_mimicry.datasets.image_loader)
get_dataset_images() (in module torch_mimicry.datasets.image_loader)
get_fake_data_images() (in module torch_mimicry.datasets.image_loader)
get_group_name() (MetricLog method)
get_imagenet_images() (in module torch_mimicry.datasets.image_loader)
get_lsun_bedroom_images() (in module torch_mimicry.datasets.image_loader)
get_random_images() (in module torch_mimicry.datasets.image_loader)
get_stl10_images() (in module torch_mimicry.datasets.image_loader)
gp_scale (WGANGPBaseDiscriminator attribute)
(WGANGPDiscriminator128 attribute)
(WGANGPDiscriminator32 attribute)
(WGANGPDiscriminator48 attribute)
(WGANGPDiscriminator64 attribute)
H
hidden_channels (DBlock attribute)
(GBlock attribute)
hinge_loss_dis() (in module torch_mimicry.modules.losses)
hinge_loss_gen() (in module torch_mimicry.modules.losses)
I
in_channels (DBlock attribute)
(DBlockOptimized attribute)
(GBlock attribute)
inception_score() (in module torch_mimicry.metrics.compute_is)
infomax_loss_scale (BaseDiscriminator attribute)
(InfoMaxGANBaseGenerator attribute)
(InfoMaxGANDiscriminator128 attribute)
(InfoMaxGANDiscriminator32 attribute)
(InfoMaxGANDiscriminator48 attribute)
(InfoMaxGANDiscriminator64 attribute)
(InfoMaxGANGenerator128 attribute)
(InfoMaxGANGenerator32 attribute)
(InfoMaxGANGenerator48 attribute)
(InfoMaxGANGenerator64 attribute)
InfoMaxGANBaseGenerator (class in torch_mimicry.nets.infomax_gan.infomax_gan_base)
InfoMaxGANDiscriminator128 (class in torch_mimicry.nets.infomax_gan.infomax_gan_128)
InfoMaxGANDiscriminator32 (class in torch_mimicry.nets.infomax_gan.infomax_gan_32)
InfoMaxGANDiscriminator48 (class in torch_mimicry.nets.infomax_gan.infomax_gan_48)
InfoMaxGANDiscriminator64 (class in torch_mimicry.nets.infomax_gan.infomax_gan_64)
InfoMaxGANGenerator128 (class in torch_mimicry.nets.infomax_gan.infomax_gan_128)
InfoMaxGANGenerator32 (class in torch_mimicry.nets.infomax_gan.infomax_gan_32)
InfoMaxGANGenerator48 (class in torch_mimicry.nets.infomax_gan.infomax_gan_48)
InfoMaxGANGenerator64 (class in torch_mimicry.nets.infomax_gan.infomax_gan_64)
infonce_loss() (BaseDiscriminator method)
items() (MetricLog method)
K
keys() (MetricLog method)
kid_score() (in module torch_mimicry.metrics.compute_kid)
L
linear_decay() (LRScheduler method)
load_celeba_dataset() (in module torch_mimicry.datasets.data_utils)
load_cifar100_dataset() (in module torch_mimicry.datasets.data_utils)
load_cifar10_dataset() (in module torch_mimicry.datasets.data_utils)
load_dataset() (in module torch_mimicry.datasets.data_utils)
load_fake_dataset() (in module torch_mimicry.datasets.data_utils)
load_from_json() (in module torch_mimicry.utils.common)
load_imagenet_dataset() (in module torch_mimicry.datasets.data_utils)
load_images() (in module torch_mimicry.utils.common)
load_lsun_bedroom_dataset() (in module torch_mimicry.datasets.data_utils)
load_stl10_dataset() (in module torch_mimicry.datasets.data_utils)
log_dir (Logger attribute)
(Trainer attribute)
log_steps (Trainer attribute)
Logger (class in torch_mimicry.training.logger)
logger (Trainer attribute)
loss_type (BaseConditionalDiscriminator attribute)
(BaseConditionalGenerator attribute)
(BaseDiscriminator attribute)
,
[1]
(BaseGenerator attribute)
(CGANPDBaseDiscriminator attribute)
(CGANPDBaseGenerator attribute)
(CGANPDDiscriminator128 attribute)
(CGANPDDiscriminator32 attribute)
(CGANPDGenerator128 attribute)
(CGANPDGenerator32 attribute)
(DCGANBaseDiscriminator attribute)
(DCGANBaseGenerator attribute)
(DCGANDiscriminator128 attribute)
(DCGANDiscriminator32 attribute)
(DCGANDiscriminator48 attribute)
(DCGANDiscriminator64 attribute)
(DCGANDiscriminatorCIFAR attribute)
(DCGANGenerator128 attribute)
(DCGANGenerator32 attribute)
(DCGANGenerator48 attribute)
(DCGANGenerator64 attribute)
(DCGANGeneratorCIFAR attribute)
(InfoMaxGANBaseGenerator attribute)
(InfoMaxGANDiscriminator128 attribute)
(InfoMaxGANDiscriminator32 attribute)
(InfoMaxGANDiscriminator48 attribute)
(InfoMaxGANDiscriminator64 attribute)
(InfoMaxGANGenerator128 attribute)
(InfoMaxGANGenerator32 attribute)
(InfoMaxGANGenerator48 attribute)
(InfoMaxGANGenerator64 attribute)
(SNGANBaseGenerator attribute)
(SNGANDiscriminator128 attribute)
(SNGANDiscriminator32 attribute)
(SNGANDiscriminator64 attribute)
(SNGANGenerator128 attribute)
(SNGANGenerator32 attribute)
(SNGANGenerator48 attribute)
(SNGANGenerator64 attribute)
(SSGANBaseDiscriminator attribute)
(SSGANBaseGenerator attribute)
(SSGANDiscriminator128 attribute)
(SSGANDiscriminator32 attribute)
(SSGANDiscriminator48 attribute)
(SSGANDiscriminator64 attribute)
(SSGANGenerator128 attribute)
(SSGANGenerator32 attribute)
(SSGANGenerator48 attribute)
(SSGANGenerator64 attribute)
(WGANGPBaseDiscriminator attribute)
(WGANGPBaseGenerator attribute)
(WGANGPDiscriminator128 attribute)
(WGANGPDiscriminator32 attribute)
(WGANGPDiscriminator48 attribute)
(WGANGPDiscriminator64 attribute)
(WGANGPGenerator128 attribute)
(WGANGPGenerator32 attribute)
(WGANGPGenerator48 attribute)
(WGANGPGenerator64 attribute)
lr_D (LRScheduler attribute)
lr_decay (LRScheduler attribute)
(Trainer attribute)
lr_G (LRScheduler attribute)
LRScheduler (class in torch_mimicry.training.scheduler)
M
MetricLog (class in torch_mimicry.training.metric_log)
metrics_dict (MetricLog attribute)
minimax_loss_dis() (in module torch_mimicry.modules.losses)
minimax_loss_gen() (in module torch_mimicry.modules.losses)
N
n_dis (Trainer attribute)
ndf (BaseConditionalDiscriminator attribute)
(BaseDiscriminator attribute)
,
[1]
(CGANPDBaseDiscriminator attribute)
(CGANPDDiscriminator128 attribute)
(CGANPDDiscriminator32 attribute)
(DCGANBaseDiscriminator attribute)
(DCGANDiscriminator128 attribute)
(DCGANDiscriminator32 attribute)
(DCGANDiscriminator48 attribute)
(DCGANDiscriminator64 attribute)
(DCGANDiscriminatorCIFAR attribute)
(InfoMaxGANDiscriminator128 attribute)
(InfoMaxGANDiscriminator32 attribute)
(InfoMaxGANDiscriminator48 attribute)
(InfoMaxGANDiscriminator64 attribute)
(SNGANBaseDiscriminator attribute)
(SNGANDiscriminator128 attribute)
(SNGANDiscriminator32 attribute)
(SNGANDiscriminator64 attribute)
(SSGANBaseDiscriminator attribute)
(SSGANDiscriminator128 attribute)
(SSGANDiscriminator32 attribute)
(SSGANDiscriminator48 attribute)
(SSGANDiscriminator64 attribute)
(WGANGPBaseDiscriminator attribute)
(WGANGPDiscriminator128 attribute)
(WGANGPDiscriminator32 attribute)
(WGANGPDiscriminator48 attribute)
(WGANGPDiscriminator64 attribute)
netD (Trainer attribute)
netD_ckpt_file (Trainer attribute)
netG (Trainer attribute)
netG_ckpt_file (Trainer attribute)
ngf (BaseConditionalGenerator attribute)
(BaseGenerator attribute)
(CGANPDBaseGenerator attribute)
(CGANPDGenerator128 attribute)
(CGANPDGenerator32 attribute)
(DCGANBaseGenerator attribute)
(DCGANGenerator128 attribute)
(DCGANGenerator32 attribute)
(DCGANGenerator48 attribute)
(DCGANGenerator64 attribute)
(DCGANGeneratorCIFAR attribute)
(InfoMaxGANBaseGenerator attribute)
(InfoMaxGANGenerator128 attribute)
(InfoMaxGANGenerator32 attribute)
(InfoMaxGANGenerator48 attribute)
(InfoMaxGANGenerator64 attribute)
(SNGANBaseGenerator attribute)
(SNGANGenerator128 attribute)
(SNGANGenerator32 attribute)
(SNGANGenerator48 attribute)
(SNGANGenerator64 attribute)
(SSGANBaseGenerator attribute)
(SSGANGenerator128 attribute)
(SSGANGenerator32 attribute)
(SSGANGenerator48 attribute)
(SSGANGenerator64 attribute)
(WGANGPBaseGenerator attribute)
(WGANGPGenerator128 attribute)
(WGANGPGenerator32 attribute)
(WGANGPGenerator48 attribute)
(WGANGPGenerator64 attribute)
nrkhs (BaseDiscriminator attribute)
(InfoMaxGANDiscriminator128 attribute)
(InfoMaxGANDiscriminator32 attribute)
(InfoMaxGANDiscriminator48 attribute)
(InfoMaxGANDiscriminator64 attribute)
ns_loss_gen() (in module torch_mimicry.modules.losses)
num_classes (BaseConditionalDiscriminator attribute)
(BaseConditionalGenerator attribute)
(CGANPDBaseDiscriminator attribute)
(CGANPDBaseGenerator attribute)
(CGANPDDiscriminator128 attribute)
(CGANPDDiscriminator32 attribute)
(CGANPDGenerator128 attribute)
(CGANPDGenerator32 attribute)
(ConditionalBatchNorm2d attribute)
(GBlock attribute)
num_epochs (Logger attribute)
num_features (ConditionalBatchNorm2d attribute)
num_steps (Logger attribute)
(LRScheduler attribute)
(Trainer attribute)
nz (BaseConditionalGenerator attribute)
(BaseGenerator attribute)
(CGANPDBaseGenerator attribute)
(CGANPDGenerator128 attribute)
(CGANPDGenerator32 attribute)
(DCGANBaseGenerator attribute)
(DCGANGenerator128 attribute)
(DCGANGenerator32 attribute)
(DCGANGenerator48 attribute)
(DCGANGenerator64 attribute)
(DCGANGeneratorCIFAR attribute)
(InfoMaxGANBaseGenerator attribute)
(InfoMaxGANGenerator128 attribute)
(InfoMaxGANGenerator32 attribute)
(InfoMaxGANGenerator48 attribute)
(InfoMaxGANGenerator64 attribute)
(SNGANBaseGenerator attribute)
(SNGANGenerator128 attribute)
(SNGANGenerator32 attribute)
(SNGANGenerator48 attribute)
(SNGANGenerator64 attribute)
(SSGANBaseGenerator attribute)
(SSGANGenerator128 attribute)
(SSGANGenerator32 attribute)
(SSGANGenerator48 attribute)
(SSGANGenerator64 attribute)
(WGANGPBaseGenerator attribute)
(WGANGPGenerator128 attribute)
(WGANGPGenerator32 attribute)
(WGANGPGenerator48 attribute)
(WGANGPGenerator64 attribute)
O
optD (LRScheduler attribute)
(Trainer attribute)
optG (LRScheduler attribute)
(Trainer attribute)
out_channels (DBlock attribute)
(DBlockOptimized attribute)
(GBlock attribute)
P
params (Trainer attribute)
print_log() (Logger method)
print_steps (Trainer attribute)
project_features() (BaseDiscriminator method)
R
restore_checkpoint() (BaseModel method)
S
sample_dataset_images() (in module torch_mimicry.datasets.image_loader)
save_checkpoint() (BaseModel method)
save_steps (Trainer attribute)
save_tensor_image() (in module torch_mimicry.utils.common)
scheduler (Trainer attribute)
SelfAttention (class in torch_mimicry.modules.layers)
SNConv2d() (in module torch_mimicry.modules.layers)
SNEmbedding() (in module torch_mimicry.modules.layers)
SNGANBaseDiscriminator (class in torch_mimicry.nets.sngan.sngan_base)
SNGANBaseGenerator (class in torch_mimicry.nets.sngan.sngan_base)
SNGANDiscriminator128 (class in torch_mimicry.nets.sngan.sngan_128)
SNGANDiscriminator32 (class in torch_mimicry.nets.sngan.sngan_32)
SNGANDiscriminator48 (class in torch_mimicry.nets.sngan.sngan_48)
SNGANDiscriminator64 (class in torch_mimicry.nets.sngan.sngan_64)
SNGANGenerator128 (class in torch_mimicry.nets.sngan.sngan_128)
SNGANGenerator32 (class in torch_mimicry.nets.sngan.sngan_32)
SNGANGenerator48 (class in torch_mimicry.nets.sngan.sngan_48)
SNGANGenerator64 (class in torch_mimicry.nets.sngan.sngan_64)
SNLinear() (in module torch_mimicry.modules.layers)
spectral_norm (DBlock attribute)
(DBlockOptimized attribute)
(GBlock attribute)
ss_loss_scale (SSGANBaseDiscriminator attribute)
(SSGANBaseGenerator attribute)
(SSGANDiscriminator128 attribute)
(SSGANDiscriminator32 attribute)
(SSGANDiscriminator48 attribute)
(SSGANDiscriminator64 attribute)
(SSGANGenerator128 attribute)
(SSGANGenerator32 attribute)
(SSGANGenerator48 attribute)
(SSGANGenerator64 attribute)
SSGANBaseDiscriminator (class in torch_mimicry.nets.ssgan.ssgan_base)
SSGANBaseGenerator (class in torch_mimicry.nets.ssgan.ssgan_base)
SSGANDiscriminator128 (class in torch_mimicry.nets.ssgan.ssgan_128)
SSGANDiscriminator32 (class in torch_mimicry.nets.ssgan.ssgan_32)
SSGANDiscriminator48 (class in torch_mimicry.nets.ssgan.ssgan_48)
SSGANDiscriminator64 (class in torch_mimicry.nets.ssgan.ssgan_64)
SSGANGenerator128 (class in torch_mimicry.nets.ssgan.ssgan_128)
SSGANGenerator32 (class in torch_mimicry.nets.ssgan.ssgan_32)
SSGANGenerator48 (class in torch_mimicry.nets.ssgan.ssgan_48)
SSGANGenerator64 (class in torch_mimicry.nets.ssgan.ssgan_64)
step() (LRScheduler method)
T
torch_mimicry.datasets.data_utils (module)
torch_mimicry.datasets.image_loader (module)
torch_mimicry.metrics.compute_fid (module)
torch_mimicry.metrics.compute_is (module)
torch_mimicry.metrics.compute_kid (module)
torch_mimicry.metrics.compute_metrics (module)
torch_mimicry.modules.layers (module)
torch_mimicry.modules.losses (module)
torch_mimicry.modules.resblocks (module)
torch_mimicry.nets.basemodel.basemodel (module)
torch_mimicry.nets.cgan_pd.cgan_pd_128 (module)
torch_mimicry.nets.cgan_pd.cgan_pd_32 (module)
torch_mimicry.nets.cgan_pd.cgan_pd_base (module)
torch_mimicry.nets.dcgan.dcgan_128 (module)
torch_mimicry.nets.dcgan.dcgan_32 (module)
torch_mimicry.nets.dcgan.dcgan_48 (module)
torch_mimicry.nets.dcgan.dcgan_64 (module)
torch_mimicry.nets.dcgan.dcgan_base (module)
torch_mimicry.nets.dcgan.dcgan_cifar (module)
torch_mimicry.nets.gan.cgan (module)
torch_mimicry.nets.gan.gan (module)
torch_mimicry.nets.infomax_gan.infomax_gan_128 (module)
torch_mimicry.nets.infomax_gan.infomax_gan_32 (module)
torch_mimicry.nets.infomax_gan.infomax_gan_48 (module)
torch_mimicry.nets.infomax_gan.infomax_gan_64 (module)
torch_mimicry.nets.infomax_gan.infomax_gan_base (module)
torch_mimicry.nets.sngan.sngan_128 (module)
torch_mimicry.nets.sngan.sngan_32 (module)
torch_mimicry.nets.sngan.sngan_48 (module)
torch_mimicry.nets.sngan.sngan_64 (module)
torch_mimicry.nets.sngan.sngan_base (module)
torch_mimicry.nets.ssgan.ssgan_128 (module)
torch_mimicry.nets.ssgan.ssgan_32 (module)
torch_mimicry.nets.ssgan.ssgan_48 (module)
torch_mimicry.nets.ssgan.ssgan_64 (module)
torch_mimicry.nets.ssgan.ssgan_base (module)
torch_mimicry.nets.wgan_gp.wgan_gp_128 (module)
torch_mimicry.nets.wgan_gp.wgan_gp_32 (module)
torch_mimicry.nets.wgan_gp.wgan_gp_48 (module)
torch_mimicry.nets.wgan_gp.wgan_gp_64 (module)
torch_mimicry.nets.wgan_gp.wgan_gp_base (module)
torch_mimicry.training.logger (module)
torch_mimicry.training.metric_log (module)
torch_mimicry.training.scheduler (module)
torch_mimicry.training.trainer (module)
torch_mimicry.utils.common (module)
train() (Trainer method)
train_step() (BaseConditionalDiscriminator method)
(BaseConditionalGenerator method)
(BaseDiscriminator method)
,
[1]
(BaseGenerator method)
(InfoMaxGANBaseGenerator method)
(SSGANBaseDiscriminator method)
(SSGANBaseGenerator method)
(WGANGPBaseDiscriminator method)
(WGANGPBaseGenerator method)
Trainer (class in torch_mimicry.training.trainer)
U
upsample (GBlock attribute)
V
vis_images() (Logger method)
vis_steps (Trainer attribute)
W
wasserstein_loss_dis() (in module torch_mimicry.modules.losses)
wasserstein_loss_gen() (in module torch_mimicry.modules.losses)
WGANGPBaseDiscriminator (class in torch_mimicry.nets.wgan_gp.wgan_gp_base)
WGANGPBaseGenerator (class in torch_mimicry.nets.wgan_gp.wgan_gp_base)
WGANGPDiscriminator128 (class in torch_mimicry.nets.wgan_gp.wgan_gp_128)
WGANGPDiscriminator32 (class in torch_mimicry.nets.wgan_gp.wgan_gp_32)
WGANGPDiscriminator48 (class in torch_mimicry.nets.wgan_gp.wgan_gp_48)
WGANGPDiscriminator64 (class in torch_mimicry.nets.wgan_gp.wgan_gp_64)
WGANGPGenerator128 (class in torch_mimicry.nets.wgan_gp.wgan_gp_128)
WGANGPGenerator32 (class in torch_mimicry.nets.wgan_gp.wgan_gp_32)
WGANGPGenerator48 (class in torch_mimicry.nets.wgan_gp.wgan_gp_48)
WGANGPGenerator64 (class in torch_mimicry.nets.wgan_gp.wgan_gp_64)
write_summaries() (Logger method)
write_to_json() (in module torch_mimicry.utils.common)
writers (Logger attribute)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.