PyTorch torch.utils.model_zoo

原文： PyTorch torch.utils.model_zoo

移至 <cite>torch.hub</cite> 。

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

將 Torch 序列化對(duì)象加載到給定的 URL。

如果下載的文件是 zip 文件，它將被自動(dòng)解壓縮。

如果 <cite>model_dir</cite> 中已經(jīng)存在該對(duì)象，則將其反序列化并返回。 <cite>model_dir</cite> 的默認(rèn)值為$TORCH_HOME/checkpoints，其中環(huán)境變量$TORCH_HOME的默認(rèn)值為$XDG_CACHE_HOME/torch。 $XDG_CACHE_HOME遵循 Linux 文件系統(tǒng)布局的 X 設(shè)計(jì)組規(guī)范，如果未設(shè)置，則默認(rèn)值為~/.cache。

參數(shù)

• url (字符串）–要下載的對(duì)象的 URL
• model_dir (字符串 ， 可選）–保存對(duì)象的目錄
• map_location (可選）–指定如何重新映射存儲(chǔ)位置的函數(shù)或命令(請(qǐng)參見(jiàn) torch.load）
• 進(jìn)度 (bool ， 可選）–是否顯示 stderr 進(jìn)度條。 默認(rèn)值：True
• check_hash (bool ， 可選）–如果為 True，則 URL 的文件名部分應(yīng)遵循命名約定filename-<sha256>.ext，其中[ <sha256>是文件內(nèi)容的 SHA256 哈希值的前 8 位或更多位。 哈希用于確保唯一的名稱并驗(yàn)證文件的內(nèi)容。 默認(rèn)值：False

例

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