torch.utils.model_zoo

torch.utils.model_zoo.load_url(url, model_dir=None)

在給定URL上加載Torch序列化對象。

如果對象已經存在於 model_dir 中，則將被反序列化並返回。URL的文件名部分應遵循命名約定filename-<sha256>.ext，其中<sha256>是文件內容的SHA256哈希的前八位或更多位數字。哈希用於確保唯一的名稱並驗證文件的內容。model_dir 的默認值爲$TORCH_HOME/models，其中$TORCH_HOME默認爲~/.torch。可以使用$TORCH_MODEL_ZOO環境變量來覆蓋默認目錄。

參數：

• url (string) - 要下載對象的URL
• model_dir (string, optional) - 保存對象的目錄

例子：

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

---