PyTorch Deep Generative Replay 项目教程

PyTorch Deep Generative Replay 项目教程

pytorch-deep-generative-replayPyTorch implementation of "Continual Learning with Deep Generative Replay", NIPS 2017项目地址:https://gitcode.com/gh_mirrors/py/pytorch-deep-generative-replay

1. 项目的目录结构及介绍

pytorch-deep-generative-replay/
├── LICENSE
├── README.md
├── const.py
├── data.py
├── dgr.py
├── gan.py
├── main.py
├── models/
├── requirements.txt
├── run_full_experiments
├── train.py
├── utils.py
└── visual.py

• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。
• const.py: 常量定义文件。
• data.py: 数据处理文件。
• dgr.py: Deep Generative Replay 实现文件。
• gan.py: GAN（生成对抗网络）实现文件。
• main.py: 项目主启动文件。
• models/: 存放模型定义的文件夹。
• requirements.txt: 项目依赖文件。
• run_full_experiments: 运行完整实验的脚本。
• train.py: 训练模型文件。
• utils.py: 工具函数文件。
• visual.py: 可视化文件。

2. 项目的启动文件介绍

main.py

main.py 是项目的启动文件，负责初始化参数、加载数据、训练模型等核心功能。以下是一些关键参数和用法：

$ python main.py --help

输出：

usage: PyTorch implementation of Deep Generative Replay [-h] [--experiment [permutated-mnist svhn-mnist mnist-svhn]] [--mnist-permutation-number MNIST_PERMUTATION_NUMBER] [--mnist-permutation-seed MNIST_PERMUTATION_SEED] --replay-mode [exact-replay generative-replay none] [--generator-z-size GENERATOR_Z_SIZE] [--generator-c-channel-size GENERATOR_C_CHANNEL_SIZE] [--generator-g-channel-size GENERATOR_G_CHANNEL_SIZE] [--solver-depth SOLVER_DEPTH] [--solver-reducing-layers SOLVER_REDUCING_LAYERS] [--solver-channel-size SOLVER_CHANNEL_SIZE] [--generator-c-updates-per-g-update GENERATOR_C_UPDATES_PER_G_UPDATE] [--generator-iterations GENERATOR_ITERATIONS]

关键参数说明

• --experiment: 指定实验类型，可选值包括 permutated-mnist, svhn-mnist, mnist-svhn。
• --replay-mode: 指定重放模式，可选值包括 exact-replay, generative-replay, none。
• --generator-z-size: 生成器中隐变量的大小。
• --solver-depth: 解决器的深度。

3. 项目的配置文件介绍

requirements.txt

requirements.txt 文件列出了项目运行所需的所有依赖包及其版本。使用以下命令安装依赖：

$ pip install -r requirements.txt

示例内容

torch==1.9.0
torchvision==0.10.0
numpy==1.21.2
matplotlib==3.4.3

其他配置文件

• const.py: 定义了项目中使用的常量，如数据路径、模型参数等。
• data.py: 包含了数据加载和预处理的配置。
• models/: 包含了模型定义的配置，如网络结构、层数等。

以上是 pytorch-deep-generative-replay 项目的基本教程，涵盖了项目的目录结构、启动文件和配置文件的介绍。希望这些信息能帮助你更好地理解和使用该项目。

pytorch-deep-generative-replayPyTorch implementation of "Continual Learning with Deep Generative Replay", NIPS 2017项目地址:https://gitcode.com/gh_mirrors/py/pytorch-deep-generative-replay