PyTorch載入模型,並輸出參數

# define the model
model = Model()
for k, v in model.named_parameters():
    print(k, v.size())
#===================================#
pretrained_state = torch.load('pretrained_model.pth')
for i in pretrained_state:
    print(i, pretrained_state[i].size())