Pytorch=1.7.1
py=3.8
cuda=102
cudnn=7
As for the input shape, it’s (batch_size, channels, fixed_height, width)
an example would be (16, 1, 96, 620) or (8, 1, 96, 702)
So essentially it’s a gray scale image with fixed height and adaptive width.
Thank you