I think you main problem is that Tensorflow input has size [708, 256, 3] with [batch, width, channels]. While pytorch expect [batch, channels, width].
So you need to be careful to give an input of size [708, 3, 256] to the pytorch model. You should check the doc for these functions to make sure you’re doing the right thing.