torch.hub.load()参数说明
时间: 2023-07-18 21:25:52 浏览: 302
基于Torch Hub的深度估计模型MiDaS-python源码.zip
`torch.hub.load()` 是 PyTorch 提供的一个方便的 API,用于从 GitHub 上加载预训练的模型和其他资源。下面是 `torch.hub.load()` 的参数说明:
- `repo_or_dir`:必需。字符串类型,指定 GitHub 上的仓库或本地目录的 URL 或路径。
- `model`:必需。字符串类型,指定要加载的模型的名称或路径。
- `source`:可选。字符串类型,指定要加载的模型的来源。默认是 `github`。
- `branch`:可选。字符串类型,指定要加载的模型的分支。默认是 `master`。
- `force_reload`:可选。布尔类型,指定是否强制重新下载模型。默认是 `False`。
- `*args` 和 `**kwargs`:可选。其他参数将传递给加载器函数。
阅读全文