torch.utils.model_zoo

    由给定URL加载Torch序列化对象。

    默认为$TORCH_HOME/models，其中$TORCH_HOME默认是。如果不需要默认目录，可以通过环境变量$TORCH_MODEL_ZOO指定其它的目录。

    • url(string）– 要下载的对象的URL链接
    • map_location(可选）– 函数或字典，指定如何重新映射存储位置(见torch.load）

    示例

    1. >>> state_dict = torch.utils.model_zoo.load_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')