一 前言
近期参与到了手写AI的车道线检测的学习中去,以此系列笔记记录学习与思考的全过程。车道线检测系列会持续更新,力求完整精炼,引人启示。所需前期知识,可以结合手写AI进行系统的学习。
二 简介
基于 Pytorch 构建模型时,我们通常会选择 nn.Module 作为模型的基类,搭配使用 Pytorch 的初始化模块 torch.nn.init,完成模型的初始化。MMEngine 在此基础上抽象出基础模块(BaseModule),让我们能够通过传参或配置文件来选择模型的初始化方式。
三 配置式初始化
为了能够更加灵活地初始化模型权重,MMEngine 抽象出了模块基类 BaseModule。模块基类继承自 nn.Module,在具备 nn.Module 基础功能的同时,还支持在构造时接受参数,以此来选择权重初始化方式。继承自 BaseModu