from torchvision import models
net = models.resnet101(pretrained=False)
print(net)
a= list(net.parameters())
print(a[0])#打印出来第一层的参数
当pretrained=True时,调用网络并调用训练好的参数,当为False时,只调用网络,而参数为随机赋值。
from torchvision import models
net = models.resnet101(pretrained=False)
print(net)
a= list(net.parameters())
print(a[0])#打印出来第一层的参数
当pretrained=True时,调用网络并调用训练好的参数,当为False时,只调用网络,而参数为随机赋值。