基于pytorch框架的lenet模型并导出ONNX格式
import torch.nn as nn
import torch.nn.functional as F
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d.
原创
2022-02-22 22:55:25 ·
649 阅读 ·
0 评论