TorchGAN：一个灵活的 GAN 训练与评估框架
机器学习
2021-11-22 v1 计算机视觉与模式识别
机器学习
摘要
TorchGAN 是一个基于 PyTorch 的框架，用于编写简洁易懂的代码以训练和评估生成对抗网络（Generative Adversarial Networks, GAN）。该框架的模块化设计允许对模型架构、损失函数、训练范式和评估指标进行轻松定制。TorchGAN 的主要特点在于其可扩展性、对大量流行模型、损失和评估指标的内置支持，以及与原生 PyTorch 相比的零开销。通过使用该框架实现若干流行的 GAN 模型，我们展示了其可扩展性和易用性。我们还针对所述模型将我们框架的训练时间与相应的基线 PyTorch 实现进行了基准测试，并观察到 TorchGAN 的特性几乎不带来任何开销。
引用
@article{arxiv.1909.03410,
title = {TorchGAN: A Flexible Framework for GAN Training and Evaluation},
author = {Avik Pal and Aniket Das},
journal= {arXiv preprint arXiv:1909.03410},
year = {2021}
}