disentangling-vae:用于了解VAE潜在表示中的解缠结的实验

【文件属性】：

文件名称：disentangling-vae:用于了解VAE潜在表示中的解缠结的实验

文件大小：166.8MB

文件格式：ZIP

更新时间：2024-05-21 13:53:35

deep-learning reproducible-research pytorch mnist chairs-dataset

解缠的VAE 该存储库包含用于研究VAE中解缠结的代码（训练/指标/绘图），并使用比较5种不同的损失（差异的）： 导致的标准VAE损失 从β-VAE ^ h β-VAE B，从 解 纠缠β-TCVAE 笔记： 经过python> = 3.6测试 经过CPU和GPU测试 目录： 我们的贡献 损失说明 安装 # clone repo pip install -r requirements.txt 跑步 使用python main.py 训练和/或评估模型。 例如： python main.py btcvae_celeba_mini -d celeba -l btcvae --lr 0.001 -b 256 -e 5 您可以使用-x 运行预定义的实验和超参数。 这些超参数可在hyperparam.ini中找到。 可