Does tvm relay support converting pytorch model with dict type input?

Troubleshooting
#1

I have a pytorch model. It needs two input which is dict type. I don’t know how to create a input list for dict type, so I just added the input name without shape(dict type does not have shape attribute). The shape list was like shape_list = [(input_name1), (input_name2)]

When I used torch.jit.trace and tried to use relay.frontend.from_pytorch(model, shape_list), the error message appeared. It showed

NotImplementedError: The following operators are not implemented: ['aten::values']

I modified my model and turned the input type dict into type tuple. It worked. However, I wonder if this has some way to compile a pytorch model with dict as input type rather than modify the original model.

#2

Relay doesn’t have dict-like data structures, so you need to convert dicts to tuples, as you already figured out.

#3

I got it. Thanks for your reply.