Skip to content

Fix Torch2Trt in Savant 0.6 extra #1114

@denisvmedyantsev

Description

@denisvmedyantsev
model = resnet.resnet18().eval().cuda()
dummy_input = torch.full((1, 3, 224, 224), 0.5, dtype=torch.float32).cuda()
model_trt = torch2trt(model, [dummy_input])

> AttributeError: 'tensorrt.tensorrt.INetworkDefinition' object has no attribute 'add_convolution'. Did you mean: 'add_convolution_nd'?

We have a sample with torch2trt - nanosam, see #1110

Consider replacing torch2trt with Torch-TensorRT

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions