d2l中保存和拿出训练后的权重值

保存训练的权重值：

主要用到：torch.save(net.state_dict(), "segmation.pth")    #可以改成自己的

把保存的参数拿出来：

主要用到：model = torch.load("segmation.pth") #就是取出保存的参数
                  net.load_state_dict(model)#net指的是构建好的神经网络，也就是需要先构建好自己的神经网络模型   通过这句代码就可以把训练好的代码对应的拿出来
                  net.to(devices[0])#这句话就是把模型搬到gpu设备上进行训练