Input data size

[jinsu35]

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