Python PyTorch help用法及代碼示例

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

用法:

torch.hub.help(github, model, force_reload=False, skip_validation=False)

參數：

• github(string) -格式為 的字符串，帶有可選標簽/分支。如果未指定 tag_name ，則假定默認分支為 main (如果存在)，否則為 master 。示例：‘pytorch/vision:0.10’

• model(string) -在 repo 的 hubconf.py 中定義的入口點名稱字符串

• force_reload(bool,可選的) -是否丟棄現有緩存並強製重新下載。默認為 False 。

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

顯示入口點 model 的文檔字符串。

示例

>>> print(torch.hub.help('pytorch/vision', 'resnet18', force_reload=True))