5.2利用模型块快速搭建复杂网络

文档摘要

5.2 利用模型块快速搭建复杂网络上一节中我们介绍了怎样定义PyTorch的模型，其中给出的示例都是用中的层来完成的。这种定义方式易于理解，在实际场景下不一定利于使用。当模型的深度非常大时候，使用定义模型结构需要向其中添加几百行代码，使用起来不甚方便。对于大部分模型结构（比如ResNet、DenseNet等），我们仔细观察就会发现，虽然模型有很多层，但是其中有很多重复出现的结构。考虑到每一层有其输入和输出，若干层串联成的”模块“也有其输入和输出，如果我们能将这些重复出现的层定义为一个”模块“，每次只需要向网络中添加对应的模块来构建模型，这样将会极大便利模型构建的过程。本节我们将以U-Net为例，介绍如何构建模型块，以及如何利用模型块快速搭建复杂模型。