代码阅读套路【1】模型结构

快速了解别人代码的结构是一个非常重要的能力，这种能力有利于后期在前人基础上进行下一步的优化，本篇博客从面向对象的角度介绍python环境下如何快速阅读他人代码

理解一个模型最关键的要素

使用pytorch书写的模型可以直接通过下面语句了解其结构。

print(model)

模型结构中最关键的是：

1. 模型输入的shape和type
2. 模型输出的shape，** 含义（生成或分类任务中的概率？生成任务中的样本？分类任务中的标签？回归任务中的得分？） ** 和type
3. 模型中间的组件（常见组件有卷积，残差，全连接）
4. 损失函数如何计算，训练目标是最大化损失函数还是最小化损失函数

可以使用下面语句查看模型类的参数有哪些：

print(model.__dict__)

面向对象的思维方式与模型

这个语句可以查看所有类的属性（可以试一下下列语句）。从返回的结果看可以知道一个python包也可以看成是一个类。

print(numpy.__dict__)
print(torch.__dict__)

这又涉及到面向对象的思考方式，类和对象是编程（乃至理解计算机世界）的一个重要概念。我们可以结合二者看类和对象的关系：对象是类的一个具体实例，例如小红是一个人，小红是对象，人是类。对于千千万万像小红一样的人，他们都有一些共同的属性与函数，他们可以通过python语句进行查看。

print(class.__dict__) # 查看class的全部属性
print(dir(class)) # 查看class的全部函数

这里可能有同学迷惑了，不是在讲读代码中的模型结构么，怎么扯到了面向对象了？答曰：面向对象的思维方式在了解一个模型结构的过程中十分重要。当你把模型看成一个类，阅读代码也变得有方向可寻：只要找到这个模型类的一些关键属性即可。