從 *.pth 文件中查看 Pytorch 權重

Question

我有一個用 Pytorch 創建的帶有權重的.pth文件。 我怎樣才能從這個文件中查看權重？

我試過這段代碼來加載和查看，但它不起作用（作為新手，我可能完全錯了）-

import torch
import torchvision.models as models

torch.save('weights\kharif_crops_final.pth')

models.load_state_dict(torch.load('weights\kharif_crops_final.pth'))
models.eval()
print(models)

Answer 1

import torch

model = torch.load('path')
print(model)

（驗證並確認）

Answer 2

PATH = 'weights\kharif_crops_final.pth' state = {'model': model.state_dict()} torch.save(state, PATH) model.load_state_dict(torch.load(PATH)['model']) # print weights for k, v in model.named_parameters(): print(k, v)