Python PyTorch download_url_to_file用法及代碼示例

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

用法:

torch.hub.download_url_to_file(url, dst, hash_prefix=None, progress=True)

參數：

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

• dst(string) -保存對象的完整路徑，例如/tmp/temporary_file

• hash_prefix(string,可選的) -如果不是 None，則 SHA256 下載的文件應以 hash_prefix 開頭。默認值：無

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

將給定 URL 處的對象下載到本地路徑。

示例

>>> torch.hub.download_url_to_file('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth', '/tmp/temporary_file')