MSDNet-GCN 项目使用教程

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

MSDNet-GCN 项目的目录结构如下：

MSDNet-GCN/
├── LICENSE
├── README.md
├── __init__.py
├── main.py
├── opts.py
├── run_me_script.sh
├── utils.py
└── models/
    └── report/

目录结构介绍

• LICENSE: 项目的许可证文件。
• README.md: 项目的说明文档。
• __init__.py: 初始化文件，用于标识该目录是一个 Python 包。
• main.py: 项目的主启动文件。
• opts.py: 配置选项处理文件。
• run_me_script.sh: 运行脚本文件。
• utils.py: 工具函数文件。
• models/: 模型相关文件目录。
  • report/: 模型报告文件目录。

2、项目的启动文件介绍

main.py

main.py 是项目的启动文件，负责初始化配置、加载数据、训练模型等核心功能。以下是 main.py 的主要功能介绍：

• 初始化配置: 通过 opts.py 处理命令行参数，设置模型训练的各种配置。
• 数据加载: 加载训练和测试数据集。
• 模型训练: 根据配置参数初始化模型，并进行训练。
• 模型评估: 在训练完成后，对模型进行评估。

使用示例

python3 main.py --model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 4 \
--msd-step 2 --msd-stepmode even --growth 6-12-24 --gpu 0

3、项目的配置文件介绍

opts.py

opts.py 文件负责处理命令行参数，定义了模型训练所需的各种配置选项。以下是一些主要的配置选项：

• 模型类型: --model，指定使用的模型类型，如 msdnet。
• 批量大小: -b，设置训练时的批量大小。
• 工作线程数: -j，设置数据加载的工作线程数。
• 数据集: cifar10 或 cifar100，指定使用的数据集。
• MSDNet 参数: --msd-blocks, --msd-base, --msd-step, --msd-stepmode, --growth，设置 MSDNet 模型的具体参数。
• GPU 设备: --gpu，指定使用的 GPU 设备。

使用示例

python3 main.py --model msdnet -b 64 -j 2 cifar10 --msd-blocks 10 --msd-base 4 \
--msd-step 2 --msd-stepmode even --growth 6-12-24 --gpu 0

通过以上配置，可以灵活地调整模型训练的参数，以适应不同的训练需求。