template | title | summary | author | tags | github-link | ||||||
---|---|---|---|---|---|---|---|---|---|---|---|
hub1 |
ResNet |
|
MegEngine Team |
|
import megengine.hub
model = megengine.hub.load('megengine/models', 'resnet18', pretrained=True)
# or any of these variants
# model = megengine.hub.load('megengine/models', 'resnet34', pretrained=True)
# model = megengine.hub.load('megengine/models', 'resnet50', pretrained=True)
# model = megengine.hub.load('megengine/models', 'resnet101', pretrained=True)
model.eval()
中文文档
English Doc