megengine.hub.pretrained

class pretrained(url)

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

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

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

返回

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

__init__(url)

方法

__init__(url)

上一页

megengine.hub.load_serialized_obj_from_url

下一页

megengine.hub.import_module