numel() 函数:返回数组中元素的个数 使用例子: params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel() print('#Params: %.1