Pytorch学习(5) —— 简单模型构建,损失函数,训练方法等1 构造模块化的神经网络2 损失函数3 训练方法4 网络模型库 torchvision.models5 模型加载与保存6 GPU加速总结
前4个博客,已经介绍了什么是Tensor,构建模型所需的基本类有哪些,这些类有哪些工作。下面,结合这些内容,通过简单的例子进行使用说明。文章目录1 构造模块化的神经网络2 损失函数3 训练方法3.1 SGD方法3.2 Adam方法3.3 简单用法4 网络模型库 torchvision.models5 模型加载与保存5.1 模型加载5.1.1 加载自带的预训练模型5.1.2 加载本地的训练模型5...