Python PyTorch load_state_dict_from_url用法及代碼示例

本文簡要介紹python語言中 torch.hub.load_state_dict_from_url 的用法。

用法:

torch.hub.load_state_dict_from_url(url, model_dir=None, map_location=None, progress=True, check_hash=False, file_name=None)

參數：

• url(string) -要下載的對象的 URL

• model_dir(string,可選的) -保存對象的目錄

• map_location(可選的) -指定如何重新映射存儲位置的函數或字典(請參閱 torch.load)

• progress(bool,可選的) -是否向 stderr 顯示進度條。默認值：真

• check_hash(bool,可選的) -如果為 True，則 URL 的文件名部分應遵循命名約定 filename-<sha256>.ext，其中 <sha256> 是文件內容的 SHA256 哈希的前八位或更多位。哈希用於確保名稱的唯一性並驗證文件的內容。默認值：假

• file_name(string,可選的) -下載文件的名稱。如果未設置，將使用來自 url 的文件名。

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

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

如果該對象已存在於 model_dir 中，則將其反序列化並返回。 model_dir 的默認值為 <hub_dir>/checkpoints，其中 hub_dir 是 get_dir() 返回的目錄。

示例

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