当前位置：网站首页>PyTorch之训练技巧

PyTorch之训练技巧

2022-08-10 05:33:00 【Vector Jason】

微调

例如：

 修改模型（将1000类改为10类输出）

model = timm.create_model('resnet34',num_classes=10,pretrained=True)
x = torch.randn(1,3,224,224)
output = model(x)
output.shape

输出为：

torch.Size([1, 10])

改变输入通道数（比如我们传入的图片是单通道的，但是模型需要的是三通道图片） 我们可以通过添加in_chans=1来改变

model = timm.create_model('resnet34',num_classes=10,pretrained=True,in_chans=1)
x = torch.randn(1,1,224,224)
output = model(x)

 

PS：关于float16与float32的区别 

 

 

 

原网站

版权声明
本文为[Vector Jason]所创，转载请带上原文链接，感谢
https://blog.csdn.net/Vectorln/article/details/125844998

边栏推荐

猜你喜欢

随机推荐