@ptrblck Cool.
I think you asked me for the input shapes:
For the tensorflow model the shape is (batch_size, hight, width, channels) which in my case is (64, 1, 66650,1)
For torch the shape is (batch_size, channels, height, width): (64, 1, 1, 66650).
Since, it is my first model in pytorch, I guess I am missing something pretty basic. Anyway, will wait to hear from you.