Preparing a Baseline Model - 1.3 English

Vitis AI Optimizer User Guide (UG1333)

Document ID
UG1333
Release Date
2020-12-17
Version
1.3 English

For simplicity, ResNet18 from torchvision is used here.

from torchvision.models.resnet import resnet18
model = resnet18(pretrained=True)