pytorch utils.model_zoo

廉子民
2023-12-01

应用

API

torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False, file_name=None)

加载序列化的torch对象。
如果对象已经存在model_dir,则直接加载。model_dir<hub_dir>/checkpoints中,而hub_dir可以通过get_dir()获取。

参数描述
url (string)
model_dir (string, optional)
map_location (optional)
progress (bool, optional)
check_hash (bool, optional)
file_name (string, optional)

参考:
https://pytorch.org/docs/stable/model_zoo.html*加粗样式**

 类似资料:

相关阅读

相关文章

相关问答