torch.utils.model_zoo

已移至 torch.hub。

torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False, file_name=None, weights_only=False)

載入給定 URL 上的 Torch 序列化物件。

如果下載的檔案是 zip 檔案，則會自動解壓縮。

如果該物件已存在於 model_dir 中，則會將其反序列化並傳回。 model_dir 的預設值是 <hub_dir>/checkpoints，其中 hub_dir 是由 get_dir() 傳回的目錄。

參數

• url (str) – 要下載物件的 URL。

• model_dir (str, optional) – 儲存物件的目錄 (可選)。

• map_location (optional) – 一個函式或字典，用於指定如何重新對應儲存位置 (請參閱 torch.load)。

• progress (bool, optional) – 是否將進度條顯示在 stderr。預設值：True。

• check_hash (bool, optional) – 如果為 True，URL 的檔名部分應遵循命名慣例 filename-<sha256>.ext，其中 <sha256> 是檔案內容 SHA256 雜湊值的前八位或更多位數。 雜湊值用於確保名稱的唯一性並驗證檔案內容。 預設值：False。

• file_name (str, optional) – 下載檔案的名稱。 如果未設定，將使用 url 中的檔名。

• weights_only (bool, optional) – 如果為 True，則只會載入權重，而不會載入複雜的 pickled 物件。 建議用於不受信任的來源。 更多細節請參考 load()。

傳回類型

Dict[str, Any]

範例

>>> state_dict = torch.hub.load_state_dict_from_url(
...     "https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth"
... )