PyTorch API - 2.0 Japanese

Vitis AI RNN ユーザー ガイド (UG1563)

Document ID
UG1563
Release Date
2022-01-20
Version
2.0 Japanese

API は、nndct/pytorch_binding/pytorch_nndct/apis.py モジュールにあります。

class torch_quantizer(): 
    def __init__(self,
                 quant_mode: str, # ['calib', 'test']
                 module: torch.nn.Module,
                 input_args: Union[torch.Tensor, Sequence[Any]] = None,
                 state_dict_file: Optional[str] = None,
                 output_dir: str = "quantize_result",
                 bitwidth: int = 8,
                 mix_bit: bool = False,
                 device: torch.device = torch.device("cuda"),
                 lstm: bool = False,
                 app_deploy: str = "CV",
                 qat_proc: bool = False,
                 custom_quant_ops: List[str] = None):

クラス torch_quantizer は、クオンタイザー オブジェクトを作成します。