from torchvision import models
net = models.resnet101(pretrained=False)
print(net)
a= list(net.parameters())
print(a[0])#打印出来第一层的参数
当pretrained=True时,调用网络并调用训练好的参数,当为False时,只调用网络,而参数为随机赋值。
本文演示了如何使用PyTorch加载ResNet101模型,并展示了如何设置预训练参数选项。通过设置pretrained参数为True或False,可以分别加载预训练权重或随机初始化的权重。
from torchvision import models
net = models.resnet101(pretrained=False)
print(net)
a= list(net.parameters())
print(a[0])#打印出来第一层的参数
当pretrained=True时,调用网络并调用训练好的参数,当为False时,只调用网络,而参数为随机赋值。
3937
3625
1885
2494

被折叠的 条评论
为什么被折叠?
