
PyTorch
北境の守卫
Good
展开
-
【PyTorch】模型部署
Backto PyTorch Index最佳实践 Best Practice2020-07-03 : C++ / TensorRT技术路径 PathsTorchScript-Libtorch / offcial C++ frontend官方推荐和主代码演进的方向The PyTorch team is betting heavily on TorchScript/libtorch as the path for going from research to production. Our i.原创 2020-07-03 14:34:59 · 872 阅读 · 0 评论 -
【PyTorch】目标检测与场景分割神器 - Detectron2
Backto PyTorch IndexDetectron2 @ 2019-10-10github $$ docsDetectron2 ,对 Detectron1 彻底重写,抛弃 Caffe2,直接基于 PyTorch (>= 1.3), 以 Mask-RCNN benchmark 为基底,全新构建。基于 PyTorch:直观,友好模块化:灵活,可扩展,能够直接在单个或多个 ...原创 2020-04-30 16:27:05 · 4453 阅读 · 0 评论 -
【PyTorch】开发、运行、部署环境
Backto PyTorch IndexVS Code + Docker办公编码使用 Windows, 编辑器 VS Code,Remote 到 Ubuntu 服务器,利用 Docker 隔离部署开发环境配置顺序Docker+VSCode配置属于自己的炼丹炉ubuntu+docker使用nvidia显卡运行pytorch...原创 2020-04-27 11:33:16 · 449 阅读 · 0 评论 -
【PyTorch】版本更新追踪
Backto PyTorch IndexRelease 1.5 @ 2020-04-23功能升级:主要的 torchvision,torchtext 和 torchaudio 库,并推出将模型从 Python API 转换为 C ++ API 等功能.工业级部署:Facebook 还和 Amazon 合作,推出了两个重磅的工具:TorchServe 模型服务框架和 TorchElasti...原创 2020-04-24 14:11:16 · 1073 阅读 · 0 评论 -
【PyTorch】常用 API 列表
Backto PyTorch Index列一些常用的 API,供快速查阅Conv2d ⊙\odot⊙ BatchNorm2d ⊙\odot⊙原创 2019-05-29 18:36:58 · 852 阅读 · 0 评论 -
【PyTorch】模型可视化之 pytorch-summary
Backto PyTorch Index我个人最喜欢的模型可视化工具, github安装 pip install torchsummary应用from torchsummary import summary...summary(your_model, input_size=(channels, H, W))info 分析每一层的类型、shape 和 参数量模型整体的...原创 2019-05-28 15:20:43 · 20858 阅读 · 7 评论 -
模型可视化工具
Backto PyTorch IndexTensorBoard(X)NN SVG: 在线交互,成图美观,表意可以,灵活性差。Netron: LaTex 生成工具,看上去很棒,暂未尝试tikz_cnn: 挺美观的,也是 LaTex 生成工具PlotNeuralNet: 不错Ref论文中卷积神经网络常见的示意图用什么软件来画比较容易? - 李佳骏的回答 - 知乎https:/...原创 2019-05-28 14:37:30 · 1901 阅读 · 0 评论 -
【PyTorch】小经验 Tips
Backto PyTorch IndexFlagstorch.backends.cudnn.benchmark = true : 让内置的 cuDNN 的 auto-tuner 自动寻找最适合当前配置的高效算法,来达到优化运行效率的问题。应该遵循以下准则:如果网络的输入数据维度或类型上变化不大,设置 torch.backends.cudnn.benchmark = true 可以...原创 2019-05-27 15:21:30 · 199 阅读 · 0 评论 -
【PyTorch】效率优化
Backto PyTorch IndexPytorch有什么节省内存(显存)的小技巧? - 郑哲东的回答 - 知乎https://www.zhihu.com/question/274635237/answer/573633662原创 2019-05-16 15:30:28 · 785 阅读 · 0 评论 -
【PyTorch】模型加载与保存
Backto PyTorch Index模型保存# 保存整个模型的结构信息和参数信息, 保存的对象是模型 model ,可以是pth方式,也可以是pkl的方式torch.save(model, './model.pth') # 方式1# 保存模型的参数, 保存的对象是模型的状态 model.state_dict()torch.save(model.state_dict(), '....原创 2019-05-15 15:28:24 · 677 阅读 · 0 评论 -
【PyTorch】自定义模型结构
Backto PyTorch Index本来想自己总结一下,发现一篇好文已经总结的非常好了:Some important Pytorch tasks - A concise summary from a vision researcher: 设计自定义模型中,freeze / update/ delete / add some layers, 和自定义 loss_function 等,非...原创 2019-05-20 15:07:58 · 730 阅读 · 0 评论 -
【DL】TensorBoardX: Bring TesorBoad to X-frameworks
Backto Deep Learning IndexGoogle 为 TensorFlow 带来了超强的可视化分析工具 TensorBoard,xmsl 其他Framework player。 TensorBoardX 应用而生,搭建起了其他 Framework(PyTorch, MXNet, numpy 等) 到 TensorBoard 的桥梁: 比如在 PyTorch 代码中 impor...原创 2019-05-08 17:22:22 · 464 阅读 · 0 评论 -
【PyTorch】模型参数初始化 weights_init
Backto PyTorch Index方法一:调用 applytorch.nn.Module.apply(fn)# 递归的调用weights_init函数,遍历nn.Module的submodule作为参数# 常用来对模型的参数进行初始化# fn是对参数进行初始化的函数的句柄,fn以nn.Module或者自己定义的nn.Module的子类作为参数# fn (Module ->...原创 2019-05-05 14:58:57 · 15035 阅读 · 1 评论 -
【DL】PyTorch Index
Backto DL Index报错场景及解决方案汇总原创 2019-03-15 11:37:05 · 486 阅读 · 0 评论 -
【PyTorch】报错场景及解决方案汇总
ErrorCaseSolutionReasonRuntimeError: Error(s) in loading state_dict for ResNet:load_state_dict()model.load_state_dict( checkpoint[‘state_dict’], strict=False)version not matchAsse...原创 2019-03-15 11:36:10 · 1917 阅读 · 0 评论