矢量量化变分自编码器
这是矢量量化变分自动编码器( )的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
1