Lua to pytorch model for forward training

Hi I have a model written in lua torch and I have convert the same model in pytorch.

But I am facing the input data feeding problem. Actually in lua the input data have both amplitude ans depth store in table format like this …

batch.data = {1:{ 1: { 1: tensor(amplitude)}, 2: { 1: tensor(depth)} } }

when I am taking the same data in my pytorch model for forward pass, it is throwing some error.

Is there something I have to change in the input data format.

What kind of error do you get and how do you pass the data?
Could you post some random input with the shapes?