Skip to content

RunTimeError for dimension specs #1870

Answered by ksivaman
harrismathew asked this question in Q&A
Discussion options

You must be logged in to vote

Even if the sizes are same you'll still have to specify shapes for both inputs, I believe something like this would work

inp_d = torch_tensorrt.Input(shape=<INSERT_SHAPE>, dtype=torch_tensorrt.dtype.float32)
module_ts = torch_tensorrt.compile(model, inputs=[inp_d, inp_d])

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by harrismathew
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants