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, 可選) – 儲存物件的目錄

• map_location (可選) – 一個函式或字典，用於指定如何重新對映儲存位置（參見 torch.load）

• progress (bool, 可選) – 是否在 stderr 中顯示進度條。預設值: True

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

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

• weights_only (bool, 可選) – 如果為 True，則僅載入權重，不載入複雜的 pickled 物件。建議用於不受信任的來源。有關更多詳細資訊，請參閱 load()。

返回型別

dict[str, Any]

示例

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