Huy Tran Van
Results
2
issues of
Huy Tran Van
@Linaom1214 I have .onnx with dynamic batch. Could you add feature to support convert that onnx to .engine model? Now this repo only support not dynamic batch. Thanks
` class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(1, 6, kernel_size=5) self.conv2 = nn.Conv2d(6, 16, kernel_size=5) self.dropout = nn.Dropout2d() self.fc1 = nn.Linear(256, 64) self.fc2 = nn.Linear(64, 10) # self.hybrid...