文件名称：vqvae:矢量量化变分自动编码器（https的pytorch实现
文件大小：158KB
文件格式：ZIP
更新时间：2024-05-24 14:51:18
JupyterNotebook
矢量量化变分自编码器 这是矢量量化变分自动编码器（ ）的PyTorch实现。 您可以找到作者的带有。 安装依赖项 要安装依赖项，请使用Python 3创建一个conda或虚拟环境，然后运行pip install -r requirements.txt 。 运行VQ VAE 要运行VQ-VAE，只需运行python3 main.py 如果要保存模型，请确保包括-save标志。 您也可以在命令行中添加参数。 默认值指定如下： parser . add_argument ( "--batch_size" , type = int , default = 32 ) parser . add_argument ( "--n_updates" , type = int , default = 5000 ) parser . add_argument ( "--n_hiddens" , type
【文件预览】：
vqvae-master
----models()
--------vqvae.py(2KB)
--------quantizer.py(2KB)
--------residual.py(2KB)
--------__init__.py(0B)
--------encoder.py(2KB)
--------decoder.py(1KB)
----utils.py(4KB)
----main.py(3KB)
----requirements.txt(167B)
----__init__.py(0B)
----datasets()
--------block.py(2KB)
--------__init__.py(0B)
----README.md(4KB)
----pixelcnn()
--------gated_pixelcnn.py(5KB)
--------__init__.py(0B)
--------models.py(4KB)
----visualization.ipynb(234KB)
----.gitignore(84B)