numel()函數:返回數組中元素的個數。
例:
params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()
print('#Params: %.1fM' % (params))
net.parameters():是Pytorch用法,用來返回net網絡中的參數
params:用來返回net網絡中的參數的總數目
numel()函數:返回數組中元素的個數。
例:
params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()
print('#Params: %.1fM' % (params))
net.parameters():是Pytorch用法,用來返回net網絡中的參數
params:用來返回net網絡中的參數的總數目
1.while循環的格式 while 條件: 條件滿足時,做的事情1 條件滿足時,做的事情2 條件滿足時,做的事情3 ...(省略)... demo