Does nn.parallel.data_parallel support multiple inputs

Hi, all

When the model has two inputs, how can I pass those two inputs into the model by using the nn.parallel.data_parallel?

The code img_feat = nn.parallel.data_parallel(model, inputs=[image_1, image_2], device_ids=range(2)) does not work. How can I make it happen?

Thanks very much.

The code is super simple, have a look:

it should be: inputs=(image_1, image_2), i.e. a tuple, not a list