arXiv reaDer
TorchGAN:GANトレーニングと評価のための柔軟なフレームワーク
TorchGAN: A Flexible Framework for GAN Training and Evaluation
  TorchGANは、Generative Adversarial Networksのトレーニングと評価のための簡潔でわかりやすいコードを記述するためのPyTorchベースのフレームワークです。このフレームワークのモジュール設計により、モデルアーキテクチャ、損失関数、トレーニングパラダイム、および評価指標を簡単にカスタマイズできます。 TorchGANの主要な機能は、その拡張性、多数の一般的なモデルの組み込みサポート、損失と評価指標、およびバニラPyTorchと比較したオーバーヘッドのゼロです。フレームワークを使用していくつかの一般的なGANモデルを実装することにより、その拡張性と使いやすさを実証します。また、対応するベースラインPyTorch実装に対する上記モデルのフレームワークのトレーニング時間のベンチマークを行い、TorchGANの機能のオーバーヘッドがほとんどゼロであることを確認します。
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.
updated: Sun Sep 08 2019 09:32:01 GMT+0000 (UTC)
published: Sun Sep 08 2019 09:32:01 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト