vmirly1:

I can train one model in PyTorch and then transform the learned weights into a TensorFlow model.


Thanks again for your quick reply. I just tried doing this (but instead transfer weights from TF model to PyTorch), and the outputs seem to be pretty far off. But that shouldn’t happen because now the models and the weights are supposedly the same. Any ideas what else could be causing this?