-
利用PyTorch-generative,您能够在PyTorch中轻松构建生成模型
资源介绍
火炬发电机
pytorch-generative是一个Python库,通过提供以下功能,可以更轻松地在PyTorch中进行生成式建模:
SOTA生成高质量参考实现
文献中常见的有用抽象
用于, 和使用实用程序
与TensorBoard集成以轻松实现指标可视化
要开始使用,请单击下面的链接之一。
安装
要安装pytorch-generative ,请克隆存储库并安装需求:
git clone https://www.github.com/EugenHota/pytorch-generative
cd pytorch-generative
pip install -r requirements.txt
安装后,运行测试以确保一切正常:
python -m unittest discover
再现结果
我们所有的模型都实现了具有所有超参数的reproduce功能,这些超参数对于再现在支持