2021年7月30日
resnet18下载与保存，转换为ONNX模型，导出 .wts 格式的权重文件
摘要： 1.download and save to 'resnet18.pth' file: import torch from torch import nn from torch.nn import functional as F import torchvision def main(): prin 阅读全文
posted @ 2021-07-30 10:36 略略略—— 阅读(2485) 评论(0) 推荐(0) 编辑