1. Learn
  2. /
  3. 课程
  4. /
  5. 使用 PyTorch 的深度学习入门

Connected

道练习

计算参数数量

深度学习模型通常拥有大量参数。参数越多,计算复杂度越高,训练时间也会更长。因此,深度学习从业者需要清楚模型包含多少参数。

在本练习中,您将先手动计算参数数量,然后使用 .numel() 方法核对您的结果。

说明 1 / 共 2 个

undefined XP
    1
    2

问题

请手动计算下述模型的参数总数。一共有多少个?可将控制台当作计算器使用。

model = nn.Sequential(nn.Linear(9, 4),
                      nn.Linear(4, 2),
                      nn.Linear(2, 1))

答案选项