albanD:

I think you main problem is that Tensorflow input has size [708, 256, 3] with [batch, width, channels] . While pytorch expect [batch, channels, width] .


But then I will get this error:
RuntimeError: Given groups=1, weight of size 252 708 5, expected input[708, 3, 256] to have 708 channels, but got 3 channels instead