nn.Module 详解：模型为什么要继承这个类
所有 PyTorch 模型都继承 nn.Module。但你可能不知道，它帮你做了参数注册、子模块管理、train/eval 切换、设备迁移四件大事。这篇把 nn.Module 的黑盒拆开。
nn.Module
和「nn.Module」相关的文章都收在这里，方便沿着同一个主题继续读下去。
标签归档
同一主题下的内容，会更容易看出思路是怎么慢慢长出来的。
全部文章
和「nn.Module」相关的文章都收在这里，方便沿着同一个主题继续读下去。
同一主题下的内容，会更容易看出思路是怎么慢慢长出来的。
所有 PyTorch 模型都继承 nn.Module。但你可能不知道，它帮你做了参数注册、子模块管理、train/eval 切换、设备迁移四件大事。这篇把 nn.Module 的黑盒拆开。