Python PyTorch load用法及代碼示例

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

用法:

torch.hub.load(repo_or_dir, model, *args, source='github', force_reload=False, verbose=True, skip_validation=False, **kwargs)

參數：

• repo_or_dir(string) -如果source 是‘github’，這應該對應於格式為repo_owner/repo_name[:tag_name] 的github repo，帶有可選的標簽/分支，例如'pytorch/vision:0.10'。如果未指定tag_name，則默認分支假定為main(如果存在)，否則為master。如果 source 是 ‘local’ 那麽它應該是本地目錄的路徑。

• model(string) -在 repo/dir 的 hubconf.py 中定義的可調用(入口點)的名稱。

• *args(可選的) -可調用 model 的相應參數。

• source(string,可選的) -‘github’或‘local’。指定如何解釋repo_or_dir。默認為‘github’。

• force_reload(bool,可選的) -是否無條件強製重新下載 github repo。如果 source = 'local' 沒有任何效果。默認為 False 。

• verbose(bool,可選的) -如果 False ，則靜音有關命中本地緩存的消息。請注意，關於首次下載的消息無法靜音。如果 source = 'local' 沒有任何效果。默認為 True 。

• skip_validation(bool,可選的) -如果 False ，torchhub 將檢查 github 參數指定的分支或提交是否正確屬於存儲庫所有者。這將向 GitHub API 發出請求；您可以通過設置 GITHUB_TOKEN 環境變量來指定非默認 GitHub 令牌。默認為 False 。

• **kwargs(可選的) -可調用 model 的相應 kwargs。

返回：

當使用給定的 *args 和 **kwargs 調用時，model 可調用的輸出。

從 github 存儲庫或本地目錄加載模型。

注意：加載模型是典型的用例，但這也可用於加載其他對象，例如分詞器、損失函數等。

如果source 是‘github’，則repo_or_dir 應該是帶有可選標簽/分支的repo_owner/repo_name[:tag_name] 形式。

如果source 是‘local’，則repo_or_dir 應該是本地目錄的路徑。

示例

>>> # from a github repo
>>> repo = 'pytorch/vision'
>>> model = torch.hub.load(repo, 'resnet50', pretrained=True)
>>> # from a local directory
>>> path = '/some/local/path/pytorch/vision'
>>> model = torch.hub.load(path, 'resnet50', pretrained=True)