随笔分类 - 自然语言处理 / 深度学习

onnx 增删改查，修改节点，删除节点，修改input，output

摘要：一、onnx 的数据类型，共有16种 elem_type: 1 --> float32 elem_type: 2 --> uint8 elem_type: 3 --> int8 elem_type: 4 --> uint16 elem_type: 5 --> int16 elem_type: 6 - 阅读全文

posted @ 2024-07-09 11:38 细雨微光阅读(3937) 评论(0) 推荐(0)

transformers、torch train demo

摘要：通过 pytorch 训练模型的逻辑： import torch.nn as nn import torch import numpy #from torch.utils.tensorboard import SummaryWriter import time vocabList = ["0","1 阅读全文

posted @ 2024-04-16 18:17 细雨微光阅读(45) 评论(0) 推荐(0)