PyTorch 1.0 中文文档:torch.hub
译者:kunwuz
torch.hub.load(github, model, force_reload=False, *args, **kwargs)
从github上加载一个带有预训练权重的模型。
参数:
- github – 必需,一个字符串对象,格式为“repo_owner/repo_name[:tag_name]”,可选 tag/branch。如果未做指定,默认的 branch 是
master
。比方说: ‘pytorch/vision[:hub]’ - model – 必须,一个字符串对象,名字在hubconf.py中定义。
- force_reload – 可选, 是否丢弃现有缓存并强制重新下载。默认是:
False
。 - *args – 可选, 可调用的
model
的相关args参数。 - **kwargs – 可选, 可调用的
model
的相关kwargs参数。
返回: | 一个有相关预训练权重的单一模型。 |
---|
torch.hub.set_dir(d)
也可以将hub_dir
设置为本地目录来保存中间模型和检查点文件。
如果未设置此参数,环境变量TORCH_HUB_DIR 会被首先搜寻,~/.torch/hub 将被创建并用作后备。
阅读全文/改进本文
PyTorch 1.0 中文文档:torch.hub相关推荐
- PyTorch 1.0 中文文档:torch.sparse
译者:hijkzzz 警告 这个API目前还处于试验阶段,可能在不久的将来会发生变化. Torch支持COO(rdinate )格式的稀疏张量,这可以有效地存储和处理大多数元素为零的张量. 稀疏张量表 ...
- PyTorch 1.0 中文文档正式接受校对 | ApacheCN
参与方式:https://github.com/apachecn/pytorch-doc-zh/blob/master/CONTRIBUTING.md 整体进度:https://github.com/ ...
- PyTorch 1.0 中文文档:torch.onnx
译者:guobaoyo 示例:从Pytorch到Caffe2的端对端AlexNet模型 这里是一个简单的脚本程序,它将一个在 torchvision 中已经定义的预训练 AlexNet 模型导出到 O ...
- PyTorch 1.0 中文文档:torch.utils.bottleneck
译者: belonHan torch.utils.bottleneck是 调试瓶颈bottleneck时首先用到的工具.它总结了python分析工具与PyTorch自动梯度分析工具在脚本运行中情况. ...
- PyTorch 1.0 中文文档:Torch 脚本
译者:keyianpai 创建 Torch 脚本代码 将追踪和脚本化结合起来 Torch 脚本语言参考 类型 表达式 语句 变量解析 python值的使用 调试 内置函数 Torch脚本是一种从PyT ...
- PyTorch 1.0 中文文档:torch.autograd
译者:gfjiangly torch.autograd 提供类和函数,实现任意标量值函数的自动微分. 它要求对已有代码的最小改变-你仅需要用requires_grad=True关键字为需要计算梯度的声 ...
- PyTorch 1.0 中文文档:torch.nn.init
译者:GeneZC torch.nn.init.calculate_gain(nonlinearity, param=None) 返回给定非线性函数的推荐的增益值.对应关系如下表: 非线性函数 增益 ...
- PyTorch 1.0 中文文档:torch.utils.model_zoo
译者:BXuan694 torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True) 由给 ...
- PyTorch 1.0 中文文档:torch.utils.data
译者:BXuan694 class torch.utils.data.Dataset 表示数据集的抽象类. 所有用到的数据集都必须是其子类.这些子类都必须重写以下方法:__len__:定义了数据集的规 ...
最新文章
- Ubuntu 16.04.1 LTS上安装电源管理系统TLP
- 一个css和js结合的下拉菜单,支持主流浏览器
- 技术人员写blog的好处
- c语言随机迷宫生成器,作为新手该如何快速上手 C++
- 三宝机器人怎么充电_巨人通力导致吉祥三宝+36号故障怪现象的又一因素
- java 反射深度克隆_C# 使用反射来实现对象的深度复制方法
- windows server 2012安装
- python爬虫 同花顺_python 爬虫--同花顺-使用代理
- SQL注入理解与防御
- 51超声波测距简易代码
- 朴素贝叶斯分类器常用的三种条件概率模型:伯努利、多项式、高斯模型
- Linux目录结构和文件、目录常用命令
- 《google软件测试之道》精彩语句摘抄
- Photoshop学习(二):换色
- 技术最好的时代,会是技术创业最好的时代吗?
- (附源码)springboot大学医学生毕业实习分配系统 毕业设计212 002
- wxpy 微信聊天机器人的实现
- 东区机房浏览器问题记录排查
- c# 中通快递对接_基于C#的快递在线下单接口免费对接方法-快递鸟
- python制作个性签名(最后有彩蛋)
热门文章
- (69)FPGA面试题-使用不同的代码实现2:1 MUX ?使用if语句
- (61)FPGA译码器实现(function)
- (34)System Verilog类的多对象详解
- echarts中toolbox位置_echarts toolbox 扩展
- python在人工智能应用锁_python实现人工智能Ai抠图功能
- do…while 第二讲
- Nginx基本数据结构之ngx_buf_t
- arm-2009q1-203-arm-none-linux-gnueabi.bin安装方法
- PCI总线特性及信号说明
- android mtk平台的fm停止搜索,【MTK平台,手机工程模式知识及方法详解】