about
TorchGAN: A Flexible Framework for GAN Training and Evaluation (arxiv.org)
4 points by avik-pal on Sep 10, 2019 | hide | past | pdf | discuss on HN

In plain words: A toolkit for building and testing generative adversarial networks—AI systems where one network creates fake data and another judges it—lets users swap in different designs, training goals, and scoring measures. Training popular versions ran at nearly the same speed as plain PyTorch code.

Abstract

TorchGAN is a PyTorch based framework for writing succinct and comprehensible code for training and evaluation of Generative Adversarial Networks. The framework's modular design allows effortless customization of the model architecture, loss functions, training paradigms, and evaluation metrics. The key features of TorchGAN are its extensibility, built-in support for a large number of popular models, losses and evaluation metrics, and zero overhead compared to vanilla PyTorch. By using the framework to implement several popular GAN models, we demonstrate its extensibility and ease of use. We also benchmark the training time of our framework for said models against the corresponding baseline PyTorch implementations and observe that TorchGAN's features bear almost zero overhead.

Avik Pal, Aniket Das
arXiv:1909.03410 · cs.LG, cs.CV, stat.ML · submitted Sep 8, 2019
abstract · pdf · html

add comment on HN