dfa-torch 项目使用教程
1. 项目的目录结构及介绍
dfa-torch/
├── Data.lua
├── README.md
├── config.lua
├── main.lua
└── ...
- Data.lua: 数据处理模块,负责加载和预处理数据集。
- README.md: 项目说明文档,包含项目的基本介绍和使用方法。
- config.lua: 配置文件,包含项目的各种配置参数。
- main.lua: 项目的启动文件,负责初始化和运行项目。
2. 项目的启动文件介绍
main.lua 是项目的启动文件,主要负责以下功能:
- 初始化配置参数。
- 加载数据集。
- 初始化模型。
- 训练模型。
- 保存训练结果。
3. 项目的配置文件介绍
config.lua 是项目的配置文件,包含以下配置参数:
- dfa gradient: 学习梯度类型,可选值为
bp
(反向传播),fa
(反馈对齐),dfa
(直接反馈对齐)。 - maxInNorm: 输入权重的最大范数。
- maxOutNorm: 输出权重的最大范数。
- accGradient: 是否累积正常和对抗梯度。
- threads: 线程数。
- type: 设备类型,可选值为
cuda
(使用 CUDA)。 - device ID: 设备 ID。
- load: 是否加载现有网络权重。
- save: 保存目录。
- dataset: 数据集类型,可选值为
MNIST
,Cifar10
,Cifar100
,STL10
,SVHN
。 - datapath: 数据集目录。
- normalization: 数据归一化类型。
- format: 图像格式,可选值为
rgb
(RGB 格式)。 - whiten: 是否白化数据。
- augment: 是否增强训练数据。
- preProcDir: 预处理数据目录。
以上是 dfa-torch
项目的基本使用教程,希望对您有所帮助。