Conversão de Código Tensorflw 1 para 2

Preciso de um especialista em Machine Learning que revise e otimize um código legado de treinamento de redes neurais (CNN) para segmentação de imagens, utilizando a arquitetura PSPNet com backbone ResNet101. O código está em TensorFlow 1.x, e desejo migrá-lo para Tensorflow 2.x com suporte completo a gpu (cuda/cudnn), earlystopping funcional e compatibilidade com colab ou ambiente local. Também busco melhorar desempenho e organização do projeto. Valor a combinar