- 博客(2)
- 资源 (6)
- 收藏
- 关注
原创 pyTorch onnx 学习(二)
添加自定义的onnx operations 在pyTorch中定义的网络图以及其运算,在onnx中不一定支持,因此,需要自定义的添加operators。如果onnx支持则可以直接使用,一下是支持的网络以及运算:add (nonzero alpha not supported)sub (nonzero alpha not supported)muldivcatmmaddmm...
2018-09-17 13:50:00 4404 2
原创 pyTorch onnx 学习(一)
官方文档pyTorch版本AlexNet模型转换到caffe2版本的模型 AlexNet模型已经训练好,来自与torchvision包,现在需要转换成onnx格式的模型,转换后的结果为alexnet.onnx:import torchimport torchvisiondummy_input = torch.randn(10, 3, 224, 224, device='cud...
2018-09-17 11:10:36 5804
pywin32-218
2016-11-02
python2.7.3
2016-11-02
A星寻路Cocos2dx3.3自动寻路代码
2015-10-16
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人