为 vai_q_pytorch 准备文件 - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文
为 vai_q_pytorch 准备下列文件。
表 1. vai_q_pytorch 的输入文件
编号 名称 描述
1 model.pth 经预训练的 PyTorch 模型,通常为 PTH 文件。
2 model.py 包含浮点模型定义的 Python 脚本。
3 校准数据集 训练数据集的子集,包含 100 到 1000 张图像。