莫烦PyTorch学习笔记(五)——模型的存取

本文主要是介绍如何对训练好的神经网络模型进行存取。 编辑器：spyderpython 1.快速搭建神经网络 这里采用上一节介绍的方法快速搭建一个小的神经网络：web def save(): # save net1 net1 = torch.nn.Sequential( torch.nn.Linear(1, 10), torch.nn.ReLU(),

>>阅读原文<<