1. 模型训练过程划分

if __name__ == '__main__':
	...

1.1. 定义过程

1.1.1. 全局参数设置

参数 作用
learning_rate 控制模型参数更新步长
device 指定模型训练使用设备(CPU或GPU)
num_epochs 指定在训练集上训练的轮数
batch_size 指定每批数据样本数
num_workers 指定加载数据集的进程
prefetch_factor 指定每个进程加载的批数

1.1.2. 模型定义

组件 作用
writer 定义tensorboard事件记录
net 定义神经网络结构
net.apply(init_weights) 模型参数初始化
criterion 定义损失函数
optimizer 定义优化

1.2. 数据集加载过程

1.2.1. Dataset类:创建数据集

1.2.2. Dataloader类:加载数据集

1.3. 训练循环

for epoch in trange(num_epochs):
	...

2. 优化分析

2.1. 定义过程

2.2. 数据集加载过程

2.3. 训练循环

2.3.1. 训练模型

2.3.2. 评估模型

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注