def inspect(…) - 2.5 日本語

この関数は、浮動小数点モデルを検査します。

def inspect(self, 
            module: torch.nn.Module, 
            input_args: Union[torch.Tensor, Tuple[Any]], 
            device: torch.device = torch.device("cuda"),
            output_dir: str = "quantize_result",
            verbose_level: int = 1,
            image_format: Optional[str] = None):

module

運用される浮動小数点モジュール

input_args

浮動小数点モジュールの実際の入力と同じ形状を持つ入力テンソル。値は乱数を取ることができます。

device

GPU または CPU 上のトレース モデル

output_dir

検査結果を格納するディレクトリ

verbose_level

画面上に表示される検査結果の詳細レベルを制御します。デフォルト値は 1 です。

0: 検査結果の出力をオフにします。

1: CPU に割り当てられる演算のサマリ レポートを出力します。

2: すべての演算のデバイス割り当てのサマリ レポートを出力します。

image_format

可視化される検査結果をエクスポートします。SVG および PNG イメージ フォーマットをサポートします。