Does nn.parallel.data_parallel support multiple inputs

1

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.

2

The code is super simple, have a look: https://github.com/pytorch/pytorch/blob/master/torch/nn/parallel/data_parallel.py#L76-L106

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