创建模型 - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

为简便起见,此处使用来自 torchvision 的 mobilenet_v2。

from torchvision.models.mobilenet import mobilenet_v2
model = mobilenet_v2(pretrained=True)