文件名称:
vdvae-jax:JAXFlax中非常深的VAE-源码
开发工具:
文件大小: 3mb
下载次数: 0
上传时间: 2021-03-12
详细说明：JAX /亚麻中非常深的VAE
从移植而来的使用和。
我试图使该实现尽可能接近原始实现。 我能够重用大部分代码，包括仍然使用PyTorch的数据输入管道。 我建议为此安装PyTorch的仅CPU版本。
使用JAX 0.2.10，亚麻0.3.0，PyTorch 1.7.1，NumPy 1.19.2进行了测试。 我还对cifar10进行了收敛训练，并使用--conv_precision=highest从本文中复制了测试ELBO值2.87，请参见下文。 如果有人要求为cifar训练有素的检查站，我将很乐意上传它们。
从本文中可以得出一些模型样本及其可视化方式：
设置
除了JAX，Flax，NumPy和PyTorch之外，此实现还取决于和 ：
pip install pillow
pip install sklearn
另外，您还必须下载数据，具体取决于要运行的数据：
./setup_
(系统自动生成,下载前可以参看下载内容)
下载文件列表
相关说明
- 本站资源为会员上传分享交流与学习,如有侵犯您的权益,请联系我们删除.
- 本站是交换下载平台，提供交流渠道，下载内容来自于网络，除下载问题外，其它问题请自行百度。
- 本站已设置防盗链，请勿用迅雷、QQ旋风等多线程下载软件下载资源，下载后用WinRAR最新版进行解压.
- 如果您发现内容无法下载，请稍后再次尝试；或者到消费记录里找到下载记录反馈给我们.
- 下载后发现下载的内容跟说明不相乎，请到消费记录里找到下载记录反馈给我们，经确认后退回积分.
- 如下载前有疑问，可以通过点击"提供者"的名字，查看对方的联系方式，联系对方咨询.
