5.3PyTorch修改模型

文档摘要

5.3 PyTorch修改模型除了自己构建PyTorch模型外，还有另一种应用场景：我们已经有一个现成的模型，但该模型中的部分结构不符合我们的要求，为了使用模型，我们需要对模型结构进行必要的修改。随着深度学习的发展和PyTorch越来越广泛的使用，有越来越多的开源模型可以供我们使用，很多时候我们也不必从头开始构建模型。因此，掌握如何修改PyTorch模型就显得尤为重要。本节我们就来探索这一问题。经过本节的学习，你将收获：如何在已有模型的基础上：修改模型若干层添加额外输入添加额外输出 5.3.1 修改模型层我们这里以PyTorch官方视觉库torchvision预定义好的模型ResNet50为例，探索如何修改模型的某一层或者某几层。