pytorch
rock4you
这个作者很懒,什么都没留下…
展开
-
pytorch转caffe2 之 onnx转caffe2报错的解决方法 ValueError: Don't know how to translate op Unsqueeze
目标:将 pytorch模型 转为 onnx模型 再转为 caffe2模型,得到两个.pb文件pytorch转onnx可通过torch.onnx._export函数实现。onnx转caffe2有两种方法:第一种方法:使用命令行参考:https://github.com/onnx/tutorials/blob/master/tutorials/OnnxCaffe2Import.ip...原创 2019-04-05 17:55:05 · 2043 阅读 · 1 评论 -
Pytorch实现MobileNetV2神经网络搭建
论文全名 MobileNetV2: Inverted Residuals and Linear Bottlenecks下载地址https://arxiv.org/abs/1801.04381v4Pytorch实现MobileNetV2神经网络搭建的代码:#coding:utf-8import torchimport torch.nn as nnimport torch....原创 2019-04-19 22:59:39 · 1684 阅读 · 0 评论