训练模型

Info

本部分内容遇到异常时查训练过程和结果QRH

初始化程序

1. 在左侧打开notebooks目录
2. 打开part1_inference_training.ipynb
3. 在第一个单元格左侧，点击箭头所示 运行 按钮

设置训练参数

1. 将C:\expiriment\titration-new\data_mo替换为训练数据目录
2. 将model2改为你需要的名字
3. 参考下表设置训练参数，或直接运行

参数	含义	取值范围
data_dir	数据集目录	str
model_name	模型名，相同名称会继续训练	str
epochs	训练轮数	非负整数
batch_size	训练的一个超参数，越大越快，也越不稳定	非负整数
learning_rate	学习率，越大越快，也越不稳定	非负整数
val_split	取多少比例样本作为验证集	\([0, 1]\)

后面的参数看代码，通常不需要改。

训练

1. 运行下一个单元格并等待结束

预期loss曲线

1. 在你认为可以终止时点击中断（在界面上方）

建议

在见到loss显著降低后，曲线拉平后10步左右就可以停下了。训练轮数过多容易导致过拟合。

通常\(\rm{loss}<1\)

通常loss不应该停在\(0.6\sim 0.7\)

1. 如果你认为还不可以终止，但训练结束了，修改epochs参数