4_5_readandsave.py

import torch
from torch import  nn

x = torch.ones(3)
print(x)
torch.save(x,'x.pt')
x2 = torch.load('x.pt')
print(x2)

y = torch.zeros(4)
print(y)
torch.save([x,y],'xy.pt')
xy_list = torch.load('xy.pt')
print(xy_list)

torch.save({'x':x,'y':y},'xy_dict.pt')
xy=torch.load('xy_dict.pt')
print(xy)

class MLP(nn.Module):
    def __init__(self):
        super(MLP,self).__init__()
        self.hidden = nn.Linear(4,3)
        self.act = nn.ReLU()
        self.output = nn.Linear(2,1)

    def forward(self,x):
        a = self.act(self.hidden(x))
        return self.output(a)
net =MLP()
print(net.state_dict())

optimizer  =torch.optim.SGD(net.parameters(),lr= 0.001,momentum=0.9)
print(optimizer.state_dict())