标签： PyTorch nn.Module

目录 一、模型结构与参数 二、模型状态与模式 三、模型保存与加载 四、设备与数据类型 五、前向传播与计算 六、参数初始化与优化 七、调试与信息 …

目录 一、什么是nn.Module 二、基础用法 2.1 自定义模型类 2.2 实例化与调用 3.1__init__() 3.2forward(…