W3Cschool
恭喜您成為首批注冊用戶
獲得88經驗值獎勵
移至 <cite>torch.hub</cite> 。
torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False)?
將 Torch 序列化對象加載到給定的 URL。
如果下載的文件是 zip 文件,它將被自動解壓縮。
如果 <cite>model_dir</cite> 中已經存在該對象,則將其反序列化并返回。 <cite>model_dir</cite> 的默認值為$TORCH_HOME/checkpoints
,其中環(huán)境變量$TORCH_HOME
的默認值為$XDG_CACHE_HOME/torch
。 $XDG_CACHE_HOME
遵循 Linux 文件系統(tǒng)布局的 X 設計組規(guī)范,如果未設置,則默認值為~/.cache
。
參數
filename-<sha256>.ext
,其中[ <sha256>
是文件內容的 SHA256 哈希值的前 8 位或更多位。 哈希用于確保唯一的名稱并驗證文件的內容。 默認值:False例
>>> state_dict = torch.hub.load_state_dict_from_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')
Copyright©2021 w3cschool編程獅|閩ICP備15016281號-3|閩公網安備35020302033924號
違法和不良信息舉報電話:173-0602-2364|舉報郵箱:jubao@eeedong.com
掃描二維碼
下載編程獅App
編程獅公眾號
聯(lián)系方式:
更多建議: