第十四章 torch.utils.model_zoo

torch.utils.model_zoo.load_url(url, model_dir=None) 在给定 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, optional) - 保存对象的目录 例子： >>> state_dict = torch.utils.model_zoo.load_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')

torch.utils.model_zoo.load_url(url, model_dir=None)在给定 URL 上加载 Torch 序列化对象。如果对象已经存在于 model_dir 中 , 则将被反序列化并返回。 URL 的文件名部分应遵循命名约定filename-<sha256>.ext, 其中 <sha256> 是文件内容的 SHA256 哈希的前八位或更多位数字。哈希用于确保唯一的名称并验证文件的内容。model_dir 的默认值为 $TORCH_HOME/models, 其中 $TORCH_HOME 默认为 ~/.torch 。可以使用$TORCH_MODEL_ZOO 环境变量来覆盖默认目录。参数 :

� url (string) - 要下载对象的 URL

1111

关注【深度学习爱好者】更多学习资源免费获取

返回值 : 对角线位置全 1, 其它位置全 0 的 2 维张量返回值类型 : Tensor例子 : >>> torch.eye(3)1 0 00 1 00 0 1[torch.FloatTensor of size 3x3]

torch.eye torch.eye(n, m=None, out=None)返回一个 2 维张量 , 对角线位置全 1, 其它位置全 0参数 :