pretrained

class pretrained(url)[源代码]

作为装饰器能够辅助从fs、s3、http等给定url上下载预训练权重

例如,我们可以按以下方式装饰一个resnet18函数

@hub.pretrained("https://url/to/pretrained_resnet18.pkl")
def resnet18(**kwargs):
返回

当被装饰的函数具有参数 pretrained=True 时,M​​egEngine则自动下载并对返回的模型填入预训练的权重。