没有合适的资源?快使用搜索试试~ 我知道了~
深度矢量量化:VQVAE,GumbelSoftmaxes和朋友
共14个文件
py:9个
md:1个
license:1个
需积分: 43 4 下载量 19 浏览量
2021-03-04
07:40:34
上传
评论
收藏 278KB ZIP 举报
温馨提示
深度矢量量化 实现VQVAE的培训代码,即具有分类潜在变量瓶颈的自动编码器,然后可以轻松地将其插入到现有基础结构中,以对离散变量序列(GPT和朋友)建模。 dvq/vqvae.py是训练脚本的入口点,可以称为一次小型训练运行,例如: cd dvq; python vqvae.py --gpus 1 --data_dir /somewhere/to/store/cifar10 这将使用CIFAR-10上的一个半小型网络来复制原始DeepMind VQVAE论文(请参阅之前的参考资料)。 此仓库上的工作仍在进行中,目前需要阅读代码并理解这些方法。 接下来,旨在重现DALL-E的结果,因为大多数代码已经到位,但是我们需要使用logit laplace分布进行训练,调整gumbel softmax超参数,并在ImageNet +上进行训练。 参考 DeepMind的 可以使用--vq_fla
资源推荐
资源详情
资源评论
收起资源包目录
deep-vector-quantization-main.zip (14个子文件)
deep-vector-quantization-main
.gitignore 1KB
requirements.txt 72B
visualize.ipynb 352KB
LICENSE 1KB
dvq
data
__init__.py 0B
cifar10.py 1KB
model
__init__.py 0B
openai_enc_dec.py 8KB
deepmind_enc_dec.py 2KB
loss.py 2KB
quantize.py 4KB
__init__.py 0B
vqvae.py 9KB
README.md 2KB
共 14 条
- 1
资源评论
余木脑袋
- 粉丝: 24
- 资源: 4596
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功