Thank you for replying me back
actually it didn’t work also, I got below error 
RuntimeError Traceback (most recent call last)
 in () 1 for i in range(65536) : ----> 2 out_put[i,:,:] = model (data_pixel2[i,:,:])
5 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py in forward(self, input) 200 _single(0), self.dilation, self.groups) 201 return F.conv1d(input, self.weight, self.bias, self.stride, --> 202 self.padding, self.dilation, self.groups) 203 204
RuntimeError: Expected 3-dimensional input for 3-dimensional weight 5 1 5, but got 2-dimensional input of size [1, 94] instead