関数インターフェース:
paddle.flops(net, input_size, custom_ops=None, print_detail=False)
パラメータの説明:
-
net (paddle.nn.Layer|paddle.static.Program) - ネットワーク インスタンス。静的ダイアグラムでは、paddle.nn.Layer または paddle.static.Program のサブクラスである必要があります。
-
input_size (リスト) - 入力テンソルのサイズ。注:batch_size=1 のみがサポートされています。
-
custom_ops (dict、オプション) – カスタム ネットワーク レイヤーの統計を実装するために使用されるディクショナリ。ディクショナリのキーはカスタムネットワーク層のクラス、値はネットワーク層のフロップをカウントする関数です関数の実装方法はサンプルコードを参照してください。このパラメータは、
net
paddle.nn.Layer の場合にのみ有効です。デフォルト値: なし。 -
print_detail (bool、オプション) – 各ネットワーク層の詳細を印刷するかどうかを制御するブール値。デフォルト値: False。
呼び出し例:
net = MobileNetV2()
paddle.flops(net, input_size=[1,3,128,64], print_detail=True)