Input data size

Hi,
I'm trying to convert your model to coreml model using the following code.

`
example_input = torch.rand(128, 40, 3)
`
`
traced_model = torch.jit.trace(model, example_input)
`

I keep getting errors about the size of example input being wrong. Can you correct me on this?
thanks 

