pretrained¶
- class pretrained(url)[源代码]¶
Decorator which helps to download pretrained weights from the given url. Including fs, s3, http(s).
例如,我们可以按以下方式装饰一个resnet18函数
@hub.pretrained("https://url/to/pretrained_resnet18.pkl") def resnet18(**kwargs):
- 返回
当被装饰的函数具有参数
pretrained=True
时,MegEngine则自动下载并对返回的模型填入预训练的权重。