牧野的博客

一 使用pytorchviz可视化 安装依赖和pytorchviz pip install graphvizpip install tochviz （或pip install git+https://github.com/szagoruyko/pytorchviz）Graphviz 是 AT&T 开发的一款开源的图形可视化软件，可以根据dot脚本语言中绘制的...

项目实现功能1. 搭建一个简单的图片分类器，完成训练和测试 2. 转换pytorch的pth模型到ONNX格式，加载ONNX并测试项目结构images目录下存放训练和测试数据集，本例使用了kaggle竞赛的猫狗数据集，统一resize到了120*120大小；TrainTestConvertOnnx.py 是训练和测试代码，包括了pth模型到onnx的转换。训练在CPU和GP...

pytorch允许把在GPU上训练的模型加载到CPU上，也允许把在CPU上训练的模型加载到GPU上。CPU->CPU,GPU->GPUtorch.load('gen_500000.pkl')GPU->CPUtorch.load('gen_500000.pkl', map_location=lambda storage, loc: storage)或：to...

CTCCTC可以生成一个损失函数，用于在序列数据上进行监督式学习，不需要对齐输入数据及标签，经常连接在一个RNN网络的末端，训练端到端的语音和文本识别系统。CTC论文地址： http://www.cs.toronto.edu/%7Egraves/icml_2006.pdfCTC网络的输入CTC网络的输入是一个样本（图像）经过网络（一般是CNN+RNN）计算后生成的特征向量（特征序列）。特征序列里各...