Preparing Files for vai_q_pytorch - 1.4.1 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-12-13
Version
1.4.1 English
Prepare the following files for vai_q_pytorch.
Table 1. Input Files for vai_q_pytorch
No. Name Description
1 model.pth Pre-trained PyTorch model, generally pth file.
2 model.py A Python script including float model definition.
3 calibration dataset A subset of the training dataset containing 100 to 1000 images.