zxgmlcj的博客

TensofFlowTensorFlow是相对高阶的机器学习库，用户可以方便地用它设计神经网络结构，而不必为了追求高效率的实现亲自写C++或CUDA代码。灵活的移植性，对不同平台PC、IOS、Android、Servering都提供支持。CaffeCaffe设计时的目标只针对于图像，没有考虑文本、语音或者时间序列的数据，因此Caffe对卷积神经网络的支持非常好，但对时间序列RNN、LSTM等...

ONNX简介Open Neural Network Exchange（ONNX，开放神经网络交换）格式，是一个用于表示深度学习模型的标准，可使模型在不同框架之间进行转移（一般用于中间部署阶段）。目前官方支持加载ONNX模型并进行推理的深度学习框架有： Caffe2, PyTorch, MXNet，ML.NET，TensorRT 和 Microsoft CNTK，并且 TensorFlow 也非...

本篇幅介绍pytorch模型转ONNX模型一、pytorch模型保存/加载有两种方式可用于保存/加载pytorch模型 1）文件中保存模型结构和权重参数 2）文件只保留模型权重.1、文件中保存模型结构和权重参数1）pytorch模型保存import torchtorch.save(selfmodel,"save.pt")2）pytorch模型加载import torchtorc...