莫煩PyTorch學習筆記(五)——模型的存取

本文主要是介紹如何對訓練好的神經網絡模型進行存取。 編輯器：spyderpython 1.快速搭建神經網絡 這裏採用上一節介紹的方法快速搭建一個小的神經網絡：web def save(): # save net1 net1 = torch.nn.Sequential( torch.nn.Linear(1, 10), torch.nn.ReLU(),

>>阅读原文<<