All modules for which code is available
- torch_mimicry.datasets.data_utils
- torch_mimicry.datasets.image_loader
- torch_mimicry.metrics.compute_fid
- torch_mimicry.metrics.compute_is
- torch_mimicry.metrics.compute_kid
- torch_mimicry.metrics.compute_metrics
- torch_mimicry.modules.layers
- torch_mimicry.modules.losses
- torch_mimicry.modules.resblocks
- torch_mimicry.nets.basemodel.basemodel
- torch_mimicry.nets.cgan_pd.cgan_pd_128
- torch_mimicry.nets.cgan_pd.cgan_pd_32
- torch_mimicry.nets.cgan_pd.cgan_pd_base
- torch_mimicry.nets.dcgan.dcgan_128
- torch_mimicry.nets.dcgan.dcgan_32
- torch_mimicry.nets.dcgan.dcgan_48
- torch_mimicry.nets.dcgan.dcgan_64
- torch_mimicry.nets.dcgan.dcgan_base
- torch_mimicry.nets.dcgan.dcgan_cifar
- torch_mimicry.nets.gan.cgan
- torch_mimicry.nets.gan.gan
- torch_mimicry.nets.infomax_gan.infomax_gan_128
- torch_mimicry.nets.infomax_gan.infomax_gan_32
- torch_mimicry.nets.infomax_gan.infomax_gan_48
- torch_mimicry.nets.infomax_gan.infomax_gan_64
- torch_mimicry.nets.infomax_gan.infomax_gan_base
- torch_mimicry.nets.sngan.sngan_128
- torch_mimicry.nets.sngan.sngan_32
- torch_mimicry.nets.sngan.sngan_48
- torch_mimicry.nets.sngan.sngan_64
- torch_mimicry.nets.sngan.sngan_base
- torch_mimicry.nets.ssgan.ssgan_128
- torch_mimicry.nets.ssgan.ssgan_32
- torch_mimicry.nets.ssgan.ssgan_48
- torch_mimicry.nets.ssgan.ssgan_64
- torch_mimicry.nets.ssgan.ssgan_base
- torch_mimicry.nets.wgan_gp.wgan_gp_128
- torch_mimicry.nets.wgan_gp.wgan_gp_32
- torch_mimicry.nets.wgan_gp.wgan_gp_48
- torch_mimicry.nets.wgan_gp.wgan_gp_64
- torch_mimicry.nets.wgan_gp.wgan_gp_base
- torch_mimicry.training.logger
- torch_mimicry.training.metric_log
- torch_mimicry.training.scheduler
- torch_mimicry.training.trainer
- torch_mimicry.utils.common