pretrained

class pretrained(url)

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

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

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

返回

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

上一页

megengine.hub.load_serialized_obj_from_url

下一页

megengine.hub.import_module