
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 · 2102 阅读 · 1 评论 -
Pytorch实现MobileNetV2神经网络搭建
论文全名 MobileNetV2: Inverted Residuals and Linear Bottlenecks 下载地址https://arxiv.org/abs/1801.04381v4 Pytorch实现MobileNetV2神经网络搭建的代码: #coding:utf-8 import torch import torch.nn as nn import torch....原创 2019-04-19 22:59:39 · 1716 阅读 · 0 评论