PyTorch モデルのパラメータの合計数をカウントするにはどうすればよいですか? model.count_params()
Keras の場合と似ています。
ベストアンサー1
PyTorch には Keras のようにパラメータの合計数を計算する関数はありませんが、パラメータ グループごとに要素の数を合計することは可能です。
pytorch_total_params = sum(p.numel() for p in model.parameters())
計算したいだけなら訓練可能パラメーター:
pytorch_total_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
回答はこの答えPyTorchフォーラム。