训练完成后,可将模型参数保存到本地,下次直接加载,避免重新训练。对于大型模型和大数据集,定时保存模型状态可防止出错。

torch.utils.data模块提供了数据加载器,将数据分批加载,避免内存不足。

nn.Sequential为顺序容器,可简化模型构建。