numel()関数:配列の要素数を返します。
例:
params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()
print('#Params: %.1fM' % (params))
net.parameters():Pytorchの使用法、ネットネットワークのパラメーターを返すために使用
params:ネットネットワーク内のパラメーターの総数を返すために使用されます
numel()関数:配列の要素数を返します。
例:
params = sum(p.numel() for p in list(net.parameters())) / 1e6 # numel()
print('#Params: %.1fM' % (params))
net.parameters():Pytorchの使用法、ネットネットワークのパラメーターを返すために使用
params:ネットネットワーク内のパラメーターの総数を返すために使用されます