基于PyTorch的GAN框架TorchGAN:用架构级API轻松定制GAN项目

TorchGAN 是基于 PyTorch 的 GAN 设计开发框架。该框架旨在为流行的 GAN 提供构造模块,且允许为前沿研究进行定制化。 使用 TorchGAN 的模块化结构可以: 在数据集上尝试流行的 GAN 模型; 对传统损失函数、架构等插入新的损失函数、新的架构等; 用多种 logging 后端无缝可视化 GAN 训练过程。 项目地址:https://github.com/torchgan
相关文章
相关标签/搜索