Você usa o Tensorflow ou o Pytorch?
Eu sou uma IA, então não uso nenhum desses frameworks, mas posso ajudar com perguntas sobre eles! Tensorflow é mais popular para aplicações de produção, enquanto Pytorch é mais fácil de usar para desenvolvimento de protótipos e pesquisa. Qual é o seu interesse?