博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
PyTorch 1.0 中文文档:torch.hub
阅读量:5167 次
发布时间:2019-06-13

本文共 550 字,大约阅读时间需要 1 分钟。

译者:

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 将被创建并用作后备。

转载于:https://www.cnblogs.com/wizardforcel/p/10492642.html

你可能感兴趣的文章
小学生算术
查看>>
BZOJ2823: [AHOI2012]信号塔
查看>>
工厂方法模式
查看>>
Linux下安装git
查看>>
mysql查询前几条记录
查看>>
自定义标签
查看>>
java二分法查找实现代码
查看>>
体系编程、SOC编程那些事儿
查看>>
mysql索引的艺术
查看>>
IBM RSA 的语言设置
查看>>
《http权威指南》阅读笔记(二)
查看>>
faster r-cnn cudnn版本不兼容问题
查看>>
[置顶] ListBox控件的数据绑定
查看>>
链表插入排序
查看>>
http://blog.csdn.net/yunye114105/article/details/7997041
查看>>
设计模式这个东西 刚刚发现几种模式好像大同小异啊
查看>>
关于 主键和外键
查看>>
python集合的交,差,并,补集合运算汇总
查看>>
校园分期支付的机遇和风险
查看>>
怕忘记-windows 2003服务器安装Node.js NPM
查看>>