megengine.hub.pretrained¶

class pretrained(url)[源代码]¶

基类：object

装饰器，用来标识预训练权重的 url，以便于载入时自动下载权重。

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

@hub.pretrained("https://url/to/pretrained_resnet18.pkl")
def resnet18(**kwargs):
    return ResNet(BasicBlock, [2, 2, 2, 2], **kwargs)

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

方法

__init__(url)

初始化方法。