Loading a PyTorch checkpoint into TFEncoderDecoderModel.