基於PyTorch的GAN框架TorchGAN:用架構級API輕鬆定製GAN項目

TorchGAN 是基於 PyTorch 的 GAN 設計開發框架。該框架旨在爲流行的 GAN 提供構造模塊,且允許爲前沿研究進行定製化。 使用 TorchGAN 的模塊化結構可以: 在數據集上嘗試流行的 GAN 模型; 對傳統損失函數、架構等插入新的損失函數、新的架構等; 用多種 logging 後端無縫可視化 GAN 訓練過程。 項目地址:https://github.com/torchgan
相關文章
相關標籤/搜索