TorchACG是一个基于Pytorch的框架,用于基于GAN的ACG应用。
torchacg的Python项目详细描述
TorchACG是一个基于Pytorch的框架,用于设计和开发基于GAN的ACG应用程序。该框架旨在提供 为流行的Gan构建模块,并允许定制 前沿研究。在
系统要求
TorchACG应该继续努力
- 所有Linux发行版都不早于Ubuntu16.04
- macOS X
- Windows 10
安装
使用pip(稳定释放):
pip install torchacg
使用pip(对于最新的主机):
^{pr2}$来源:
git clone https://github.com/tczhangzhi/torchacg.git cd torchacg python setup.py install
入门
流行的GANs和最新的GAN网络结构 神经式传输方法是现成的:
from torchacg.model.acgan import Generator, Discriminator g_model = Generator() d_model = Discriminator()
多样化的损失函数与复杂的对抗训练方法 有不同的学习者。方便用户 在不知情的情况下快速复制最先进的培训方法 实施细节:
from torchacg.trainer.acgan import ACGANTrainer trainer = ACGANTrainer(g_model, d_model, dataloader, device=device) trainer.start(epoch=10)
试验
设置环境:
export PYTHONPATH=./
运行测试用例:
python tests/test_acgan.py
文件
马上就来。在
- 项目
标签: