PyTorch 1.0 中文文档：torch.utils.model

译者：BXuan694

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

由给定URL加载Torch序列化对象。

如果该对象已经存在于model_dir中，将被反序列化并返回。URL的文件名部分应该遵循约定filename-<sha256>.ext，其中<sha256>是文件内容的SHA256哈希的前八位或更多位数。（哈希用于确保唯一的名称并验证文件的内容）

model_dir默认为$TORCH_HOME/models，其中$TORCH_HOME默认是~/.torch。如果不需要默认目录，可以通过环境变量$TORCH_MODEL_ZOO指定其它的目录。

参数：

url（string）– 要下载的对象的URL链接
model_dir（string , 可选）– 保存下载对象的目录
map_location（可选）– 函数或字典，指定如何重新映射存储位置（见torch.load）
progress（bool, 可选）– 是否向标准输出展示进度条

示例

>>> state_dict = torch.utils.model_zoo.load_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')

阅读全文／改进本文

PyTorch 1.0 中文文档：torch.utils.model_zoo相关推荐

PyTorch 1.0 中文文档正式接受校对 | ApacheCN
参与方式:https://github.com/apachecn/pytorch-doc-zh/blob/master/CONTRIBUTING.md 整体进度:https://github.com/ ...
PyTorch 1.0 中文文档：torchvision.models
译者:BXuan694 models子包定义了以下模型架构: AlexNet VGG ResNet SqueezeNet DenseNet Inception v3 你可以通过调用以下构造函数构造随机 ...
PyTorch 1.0 中文文档：torch.utils.bottleneck
译者: belonHan torch.utils.bottleneck是调试瓶颈bottleneck时首先用到的工具.它总结了python分析工具与PyTorch自动梯度分析工具在脚本运行中情况. ...
PyTorch 1.0 中文文档：torch.utils.data
译者:BXuan694 class torch.utils.data.Dataset 表示数据集的抽象类. 所有用到的数据集都必须是其子类.这些子类都必须重写以下方法:__len__:定义了数据集的规 ...
PyTorch 1.0 中文文档：torch.utils.checkpoint
译者: belonHan 注意 checkpointing的实现方法是在向后传播期间重新运行已被checkpint的前向传播段. 所以会导致像RNG这类(模型)的持久化的状态比实际更超前.默认情况下, ...
PyTorch 1.0 中文文档：torch.utils.dlpack
译者:kunwuz torch.utils.dlpack.from_dlpack(dlpack) → Tensor 将DLPack解码成Tensor张量. 参数: dlpack – 一个有着dlten ...
PyTorch 1.0 中文文档：torch.utils.cpp_extension
译者: belonHan torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs) 创建一个C++的setuptoo ...
PyTorch 1.0 中文文档：torch.sparse
译者:hijkzzz 警告这个API目前还处于试验阶段,可能在不久的将来会发生变化. Torch支持COO(rdinate )格式的稀疏张量,这可以有效地存储和处理大多数元素为零的张量. 稀疏张量表 ...
PyTorch 1.0 中文文档：torchvision.datasets
译者:BXuan694 所有的数据集都是torch.utils.data.Dataset的子类, 即:它们实现了__getitem__和__len__方法.因此,它们都可以传递给torch.utils ...

PyTorch 1.0 中文文档：torch.utils.model_zoo

PyTorch 1.0 中文文档：torch.utils.model_zoo相关推荐

最新文章

热门文章