文件名称:mvae:混合曲率可变自动编码器(ICLR 2020)
文件大小:213KB
文件格式:ZIP
更新时间:2024-06-13 11:43:04
machine-learning deep-learning thesis geometry evaluation
混合曲率变分自动编码器 PyTorch实施 概述 该存储库包含介绍的混合曲率变分自动编码器(M-VAE)的PyTorch实现。 有关arXiv的论文,请参见: : 。 安装 安装Python 3.7+。 要安装所有依赖项,请确保已安装并运行 make conda conda activate pt make download_data 结构 chkpt/ -训练模型的检查点。 data/ -数据文件夹。 包含下载数据集和下载的数据所需的脚本。 mathematica/ -Mathematica脚本(各种公式的推导等)。 mt/ -源文件夹(代表硕士论文)。 data/ -数据加载,预处理,批处理和预训练的嵌入。 examples/ -包含主要的可执行文件。 读取标志并运行相应的训练和/或评估。 mvae/ -模型目录。 请注意,模型大量使用继承! test_data/ -