Converter código python para TensorFlow 2.1 automaticamente

Muitas vezes encontro repositorios com projetos que atendem a necessidade porém com versão do TF um pouco antiga, que é um problema para meu setup. Tenho uma placa de vídeo que roda pelo menos o CUDA 10.1 que requisita o TF2.1+.

Para nossa alegria, a equipe do TF disponibilizou uma função que analisa e converte todos seus arquivos de um diretório.

tf_upgrade_v2 --intree inputFolder/ --outtree outputFolder/ --reportfile log.txt

Assim é muito simples 🙂

Deixe uma resposta

O seu endereço de e-mail não será publicado. Campos obrigatórios são marcados com *