使用MindSpore查看模型参数

 时间:2026-04-30 10:52:35

1、可以直接加载CheckPoint统计,可能额外统计了动量和optimizer中的变量,需要过滤下相关变量。 可以参考如下接口统计网络参数量。

def count_params(net):

    """Count number of parameters in the network

    Args:

        net (mindspore.nn.Cell): Mindspore network instance

    Returns:

        total_params (int): Total number of trainable params

    """

    total_params = 0

    for param in net.trainable_params():

        total_params += np.prod(param.shape)

    return total_params

  • UltraEdit怎么设置打印时文本自动换行
  • 波尔多酒区六大酒庄
  • 离婚后如何调整心态
  • 在职场如何进行自我激励?
  • IOS版Pillow如何授权麦克风?
  • 热门搜索
    上海旅游集散中心官方网站 缙云旅游 夏季旅游胜地 芜湖旅行社旅游报价 菊花岛旅游 旅游业营改增 惠州有什么旅游景点 重庆到四姑娘山旅游 春节厦门旅游攻略 丽江最佳旅游时间