使用MindSpore查看模型参数

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