import torch
import pandas as pd
import numpy as np
import torchvision.models as models
resnet18 = models.resnet18(pretrained=True)
parm={}
for name,parameters in resnet18.named_parameters():
print(name,':',parameters.size())
parm[name]=parameters.detach().numpy()
获取Pytorch中间某一层权重
最新推荐文章于 2022-10-23 21:47:29 发布